diff --git a/neoforge-datagen/src/main/java/dev/compactmods/machines/datagen/tags/ItemTagGenerator.java b/neoforge-datagen/src/main/java/dev/compactmods/machines/datagen/tags/ItemTagGenerator.java
index 5473d613..17dcfb3b 100644
--- a/neoforge-datagen/src/main/java/dev/compactmods/machines/datagen/tags/ItemTagGenerator.java
+++ b/neoforge-datagen/src/main/java/dev/compactmods/machines/datagen/tags/ItemTagGenerator.java
@@ -42,7 +42,7 @@ private void upgrades() {
     private void shrinkingDevices(PersonalShrinkingDevice psd) {
         final var cmShrinkTag = tag(PSDTags.ITEM);
         cmShrinkTag.add(psd);
-        cmShrinkTag.add(ResourceKey.create(net.minecraft.core.registries.Registries.ITEM, new ResourceLocation("shrink", "shrinking_device")));
+        cmShrinkTag.addOptional(new ResourceLocation("shrink", "shrinking_device"));
     }
 
     private void curiosTags(PersonalShrinkingDevice psd) {