diff --git a/patches/minecraft/net/minecraft/entity/EntityClassification.java.patch b/patches/minecraft/net/minecraft/entity/EntityClassification.java.patch
index 4e6d38a83..f81882afb 100644
--- a/patches/minecraft/net/minecraft/entity/EntityClassification.java.patch
+++ b/patches/minecraft/net/minecraft/entity/EntityClassification.java.patch
@@ -13,7 +13,7 @@
        return this.field_82707_i;
     }
  
-+   public static EntityClassification create(String name, String p_i50381_3_, int p_i50381_4_, boolean p_i50381_5_, boolean p_i50381_6_) {
++   public static EntityClassification create(String name, String id, int maxNumberOfCreatureIn, boolean isPeacefulCreatureIn, boolean isAnimalIn, int despawnDistance) {
 +      throw new IllegalStateException("Enum not extended");
 +   }
 +
diff --git a/src/test/java/net/minecraftforge/debug/block/CustomPlantTypeTest.java b/src/test/java/net/minecraftforge/debug/block/CustomPlantTypeTest.java
index a5fe0a7c6..0997d7d45 100644
--- a/src/test/java/net/minecraftforge/debug/block/CustomPlantTypeTest.java
+++ b/src/test/java/net/minecraftforge/debug/block/CustomPlantTypeTest.java
@@ -87,7 +87,7 @@ public class CustomPlantTypeTest
 
     public static class CustomPlantBlock extends FlowerBlock implements IPlantable
     {
-        public static PlantType pt = PlantType.get("CustomPlantType");
+        public static PlantType pt = PlantType.get("custom_plant_type");
 
         public CustomPlantBlock()
         {
diff --git a/src/test/java/net/minecraftforge/debug/entity/CreateEntityClassificationTest.java b/src/test/java/net/minecraftforge/debug/entity/CreateEntityClassificationTest.java
new file mode 100644
index 000000000..d9ae553a1
--- /dev/null
+++ b/src/test/java/net/minecraftforge/debug/entity/CreateEntityClassificationTest.java
@@ -0,0 +1,10 @@
+package net.minecraftforge.debug.entity;
+
+import net.minecraft.entity.EntityClassification;
+import net.minecraftforge.fml.common.Mod;
+
+@Mod("create_entity_classification_test")
+public class CreateEntityClassificationTest
+{
+    public static EntityClassification test = EntityClassification.create("TEST", "test", 1, true, true, 128);
+}
diff --git a/src/test/resources/META-INF/mods.toml b/src/test/resources/META-INF/mods.toml
index 31d64569b..e6b720cd4 100644
--- a/src/test/resources/META-INF/mods.toml
+++ b/src/test/resources/META-INF/mods.toml
@@ -1,8 +1,6 @@
 modLoader="javafml"
 loaderVersion="[28,)"
 
-[[mods]]
-    modId="mark_dimension_for_deletion_test"
 [[mods]]
 	modId="containertypetest"
 [[mods]]
@@ -68,3 +66,5 @@ loaderVersion="[28,)"
     modId="stencil_enable_test"
 [[mods]]
     modId="deferred_registry_test"
+[[mods]]
+    modId="create_entity_classification_test"
\ No newline at end of file