[MC-241159] Other stones than the regular stone can spawn near lava lakes in surface in superflat world

The bug

Other stones than the regular stone can spawn near lava lakes in the surface. They should be in the depth. I don't know if it's WAI but it's strange.

How to repoduce

Create a superflat word with "Overworld" preset
Flying around the world and you will notice that there are some (deepslate) ores, tuff, diorite etc... near lava lakes.

Cause

The reason why this happened it's because it's depending of the height of the world. For example : when you create a superflat world with y=0 as the surface, you can find deepslate, tuff, rare ores, diorite, andesite, granite near lava lakes but when you change it and with y=71 as the surface, you can find coal ore near lava pool.|