BDS Client Side Chunk Generation doesn't work with custom flat worlds

When using a world with custom flat world layers on a realm or bds, the client generates normal grass flat world chunks instead of the custom flat world layers. But updates the chunks near to the proper blocks.