diff --git a/DataGenerator/src/main/java/net/minestom/datagen/DataGenType.java b/DataGenerator/src/main/java/net/minestom/datagen/DataGenType.java
index 1c0e858..348f541 100644
--- a/DataGenerator/src/main/java/net/minestom/datagen/DataGenType.java
+++ b/DataGenerator/src/main/java/net/minestom/datagen/DataGenType.java
@@ -15,7 +15,6 @@ public enum DataGenType {
     BLOCKS("blocks", new BlockGenerator()),
     COMMAND_ARGUMENTS("command_arguments", new CommandArgumentGenerator()),
     CUSTOM_STATISTICS("custom_statistics", new CustomStatisticGenerator()),
-    DIMENSION_TYPES("dimension_types", new DimensionTypeGenerator()),
     DYE_COLORS("dye_colors", new DyeColorGenerator()),
     ENCHANTMENTS("enchantments", new EnchantmentGenerator()),
     ENTITIES("entities", new EntityGenerator()),
@@ -39,9 +38,11 @@ public enum DataGenType {
     GAMEEVENT_TAGS("tags/gameplay_tags", new GameEventTagGenerator()),
     ITEM_TAGS("tags/item_tags", new ItemTagGenerator()),
 
+    DIMENSION_TYPES("dimension_types", new GenericResourceGenerator("dimension_type")),
+    CHAT_TYPES("chat_types", new GenericResourceGenerator("chat_type")),
     DAMAGE_TYPES("damage_types", new GenericResourceGenerator("damage_type")),
-    BANNER_PATTERNS("banner_patterns", new GenericResourceGenerator("banner_pattern", List.of("asset_id"))),
-    WOLF_VARIANTS("wolf_variants", new GenericResourceGenerator("wolf_variant", List.of("angry_texture", "tame_texture", "wild_texture"))),
+    BANNER_PATTERNS("banner_patterns", new GenericResourceGenerator("banner_pattern")),
+    WOLF_VARIANTS("wolf_variants", new GenericResourceGenerator("wolf_variant")),
     TRIM_MATERIALS("trim_materials", new GenericResourceGenerator("trim_material")),
     TRIM_PATTERNS("trim_patterns", new GenericResourceGenerator("trim_pattern")),
 
diff --git a/DataGenerator/src/main/java/net/minestom/generators/BiomeGenerator.java b/DataGenerator/src/main/java/net/minestom/generators/BiomeGenerator.java
index 07493ca..22c55f9 100644
--- a/DataGenerator/src/main/java/net/minestom/generators/BiomeGenerator.java
+++ b/DataGenerator/src/main/java/net/minestom/generators/BiomeGenerator.java
@@ -2,6 +2,10 @@
 
 import com.google.gson.Gson;
 import com.google.gson.JsonObject;
+import net.minecraft.core.RegistryAccess;
+import net.minecraft.core.registries.BuiltInRegistries;
+import net.minecraft.core.registries.Registries;
+import net.minecraft.data.Main;
 import net.minecraft.world.level.biome.*;
 import net.minestom.datagen.DataGenerator;
 import net.minestom.utils.ResourceUtils;
diff --git a/DataGenerator/src/main/java/net/minestom/generators/tags/BlockTagGenerator.java b/DataGenerator/src/main/java/net/minestom/generators/tags/BlockTagGenerator.java
index 2676cef..3fe501c 100644
--- a/DataGenerator/src/main/java/net/minestom/generators/tags/BlockTagGenerator.java
+++ b/DataGenerator/src/main/java/net/minestom/generators/tags/BlockTagGenerator.java
@@ -1,6 +1,9 @@
 package net.minestom.generators.tags;
 
 import com.google.gson.JsonObject;
+import net.minecraft.core.registries.BuiltInRegistries;
+import net.minecraft.server.Main;
+import net.minecraft.tags.TagEntry;
 import net.minestom.datagen.DataGenerator;
 
 public final class BlockTagGenerator extends DataGenerator {