diff --git a/src/main/java/net/krituximon/stalinium/datagen/ModItemModelProvider.java b/src/main/java/net/krituximon/stalinium/datagen/ModItemModelProvider.java
index dc3347b..818fc97 100644
--- a/src/main/java/net/krituximon/stalinium/datagen/ModItemModelProvider.java
+++ b/src/main/java/net/krituximon/stalinium/datagen/ModItemModelProvider.java
@@ -16,6 +16,10 @@ public class ModItemModelProvider extends ItemModelProvider {
         basicItem(ModItems.STALINIUM_INGOT.get());
         basicItem(ModItems.SOVIET_ANTHEM_MUSIC_DISC.get());
         basicItem(ModItems.STALINIUM_NUGGET.get());
+        basicItem(ModItems.STALINIUM_HELMET.get());
+        basicItem(ModItems.STALINIUM_CHESTPLATE.get());
+        basicItem(ModItems.STALINIUM_LEGGINGS.get());
+        basicItem(ModItems.STALINIUM_BOOTS.get());
         handheldItem(ModItems.STALINIUM_SWORD.get());
         handheldItem(ModItems.STALINIUM_AXE.get());
         handheldItem(ModItems.STALINIUM_SHOVEL.get());
diff --git a/src/main/java/net/krituximon/stalinium/datagen/ModItemTagProvider.java b/src/main/java/net/krituximon/stalinium/datagen/ModItemTagProvider.java
index dc4b33d..4bb4b65 100644
--- a/src/main/java/net/krituximon/stalinium/datagen/ModItemTagProvider.java
+++ b/src/main/java/net/krituximon/stalinium/datagen/ModItemTagProvider.java
@@ -36,6 +36,15 @@ public class ModItemTagProvider extends ItemTagsProvider {
                 .add(ModItems.STALINIUM_HOE.get());
         this.tag(ItemTags.PICKAXES)
                 .add(ModItems.STALINIUM_PICKAXE.get());
+        this.tag(ItemTags.HEAD_ARMOR)
+                .add(ModItems.STALINIUM_HELMET.get());
+        this.tag(ItemTags.CHEST_ARMOR)
+                .add(ModItems.STALINIUM_CHESTPLATE.get());
+        this.tag(ItemTags.LEG_ARMOR)
+                .add(ModItems.STALINIUM_LEGGINGS.get());
+        this.tag(ItemTags.FOOT_ARMOR)
+                .add(ModItems.STALINIUM_BOOTS.get());
+
         this.tag(ItemTags.TRIMMABLE_ARMOR)
                 .add(ModItems.STALINIUM_HELMET.get())
                 .add(ModItems.STALINIUM_CHESTPLATE.get())