diff --git a/common/addons/generation-stage-feature/src/main/java/com/dfsek/terra/addons/generation/feature/FeatureGenerationAddon.java b/common/addons/generation-stage-feature/src/main/java/com/dfsek/terra/addons/generation/feature/FeatureGenerationAddon.java
index 74eed4f98..ac0475d7e 100644
--- a/common/addons/generation-stage-feature/src/main/java/com/dfsek/terra/addons/generation/feature/FeatureGenerationAddon.java
+++ b/common/addons/generation-stage-feature/src/main/java/com/dfsek/terra/addons/generation/feature/FeatureGenerationAddon.java
@@ -87,7 +87,7 @@ public class FeatureGenerationAddon implements AddonInitializer {
                 
                         featureGenerationStages.forEach(stage -> features.put(stage, template.get(stage.getID(), List.class)));
                 
-                        event.getLoadedObject(Biome.class).getContext().put(new BiomeFeatures(features));
+                        event.getLoadedObject(Biome.class).getContext().put(biomeFeaturesKey, new BiomeFeatures(features));
                     }
                 })
                 .failThrough();