Lake features in 1.21

Hoping someone else has come across this issue and can at least confirm it, maybe explain it, or hopefully provide a solution.
I'm updating my Dimensions pack to 1.21 from 1.20.5, and where previously I was able to use this placed feature to generate a lake configured feature in basically large blobs, in 1.21 it causes a crash. Unfortunately, the crash report/output log doesn't provide any useful details and just says some variation of "Chunk not available during world generation." (Which I've dealt with already a few times when fixing other placed features)
  "feature": "nitdim:lake_water",
  "placement": [
    {
      "type": "minecraft:noise_based_count",
      "noise_to_count_ratio": 25,
      "noise_factor": 100,
      "noise_offset": -0.3
    },
    {
      "type": "minecraft:count",
      "count": 20
    },
    {
      "type": "minecraft:in_square"
    },
    {
      "type": "minecraft:heightmap",
      "heightmap": "WORLD_SURFACE_WG"
    },
    {
      "type": "minecraft:block_predicate_filter",
      "predicate": {
        "type": "minecraft:matching_blocks",
        "offset": [
          0,
          -1,
          0
        ],
        "blocks": "minecraft:grass_block"
      }
    },
    {
      "type": "minecraft:biome"
    }
  ]
}

I've tried a number of things to stop this crash, including but not limited to:

Removing the count
Removing the block_predicate_filter
Moving the in_square
Removing both counts

All still resulting in the same crash. The only thing that seemed to stop the crash was changing the noise_count_ratio to 1 with no additional count placement, but this of course doesn't produce the density of lakes that I want. What's also interesting is I use almost these exact same placement types for another feature (trees), with no issues.
This leads me to believe it's some kind of problem with Lakes specifically, since they have always had unusual behavior anyway.