diff --git a/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java b/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
index a26554c..e37fa0a 100755
--- a/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
+++ b/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
@@ -1,8 +1,14 @@
 package dev.micle.totemofreviving.data.client;
 
 import dev.micle.totemofreviving.TotemOfReviving;
-import dev.micle.totemofreviving.item.StrawChargeItem;
-import dev.micle.totemofreviving.item.StrawTotemItem;
+import dev.micle.totemofreviving.item.charge.DiamondChargeItem;
+import dev.micle.totemofreviving.item.charge.IronChargeItem;
+import dev.micle.totemofreviving.item.charge.NetheriteChargeItem;
+import dev.micle.totemofreviving.item.charge.StrawChargeItem;
+import dev.micle.totemofreviving.item.totem.DiamondTotemItem;
+import dev.micle.totemofreviving.item.totem.IronTotemItem;
+import dev.micle.totemofreviving.item.totem.NetheriteTotemItem;
+import dev.micle.totemofreviving.item.totem.StrawTotemItem;
 import net.minecraft.data.DataGenerator;
 import net.minecraftforge.client.model.generators.ItemModelBuilder;
 import net.minecraftforge.client.model.generators.ItemModelProvider;
@@ -19,7 +25,14 @@ public class ModItemModelProvider extends ItemModelProvider {
         ModelFile itemGenerated = getExistingFile(mcLoc("item/generated"));
 
         builder(itemGenerated, StrawTotemItem.getName());
+		builder(itemGenerated, IronTotemItem.getName());
+		builder(itemGenerated, DiamondTotemItem.getName());
+		builder(itemGenerated, NetheriteTotemItem.getName());
+		
 		builder(itemGenerated, StrawChargeItem.getName());
+		builder(itemGenerated, IronChargeItem.getName());
+		builder(itemGenerated, DiamondChargeItem.getName());
+		builder(itemGenerated, NetheriteChargeItem.getName());
     }
 
     private ItemModelBuilder builder(ModelFile itemGenerated, String name) {