diff --git a/core/src/mindustry/mod/ContentParser.java b/core/src/mindustry/mod/ContentParser.java
index 5f20b73550..5238fa0f15 100644
--- a/core/src/mindustry/mod/ContentParser.java
+++ b/core/src/mindustry/mod/ContentParser.java
@@ -445,10 +445,10 @@ public class ContentParser{
             UnitType unit;
             if(locate(ContentType.unit, name) == null){
 
-                unit = make(resolve(value.getString("class", ""), UnitType.class), mod + "-" + name);
+                unit = make(resolve(value.getString("template", ""), UnitType.class), mod + "-" + name);
 
-                if(value.has("class")){
-                    value.remove("class");
+                if(value.has("template")){
+                    value.remove("template");
                 }
 
                 var typeVal = value.get("type");