diff --git a/src/main/java/ivorius/psychedelicraft/world/gen/PSWorldGen.java b/src/main/java/ivorius/psychedelicraft/world/gen/PSWorldGen.java
index bdae89d3..6b8d3b11 100644
--- a/src/main/java/ivorius/psychedelicraft/world/gen/PSWorldGen.java
+++ b/src/main/java/ivorius/psychedelicraft/world/gen/PSWorldGen.java
@@ -80,7 +80,7 @@ private static void registerTilledPatch(String id, CannabisPlantBlock crop, bool
                             BiomeSelector.COLD
                             .or(BiomeSelectors.tag(BiomeTags.IS_HILL))
                             .or(BiomeSelectors.tag(BiomeTags.IS_FOREST))
-                            .or(ctx -> ctx.getBiomeKey() == BiomeKeys.PLAINS)
+                            .or(BiomeSelectors.includeByKey(BiomeKeys.PLAINS))
                     )),
                     GenerationStep.Feature.VEGETAL_DECORATION,
                     placement