In 1.21, Mojang made slight changes to world generation to improve performance. Specifically, a change with how chunks are handled.

This causes the Lake feature to crash upon worldgen, as it checks in these now unloaded chunks for the biome, in order to determine if Water it places shall freeze or not.

This mod simply removes that check, therefore fixing the crash.