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 ea13d9a..dcac200 100755
--- a/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
+++ b/src/main/java/dev/micle/totemofreviving/data/client/ModItemModelProvider.java
@@ -1,6 +1,7 @@
 package dev.micle.totemofreviving.data.client;
 
 import dev.micle.totemofreviving.TotemOfReviving;
+import dev.micle.totemofreviving.item.StrawTotemItem;
 import net.minecraft.data.DataGenerator;
 import net.minecraftforge.client.model.generators.ItemModelBuilder;
 import net.minecraftforge.client.model.generators.ItemModelProvider;
@@ -16,10 +17,7 @@ public class ModItemModelProvider extends ItemModelProvider {
     protected void registerModels() {
         ModelFile itemGenerated = getExistingFile(mcLoc("item/generated"));
 
-        builder(itemGenerated, "totem_of_reviving");
-        builder(itemGenerated, "reviving_charge");
-        builder(itemGenerated, "straw_totem");
-        builder(itemGenerated, "straw_charge");
+        builder(itemGenerated, StrawTotemItem.getName());
     }
 
     private ItemModelBuilder builder(ModelFile itemGenerated, String name) {