Solved Flatworld flatter generation

So i want to generate a flatworld that has only one layer of dirt, instead of three.

I came up with a method that fires whenever the chunk is being created, and to set the world generation accordingly. It works, but only sometimes. There are still a lot of chunks out there that generate normally.

So, instead of derping with the chunk load event, is there an easier way to make the creative worlds spawn with only a single layer of grass over the bedrock?

cuz this is what happens with my code:
edit: whatevs, i just made a custom generator anyways