diff --git a/src/main/java/dev/wolfieboy09/qstorage/datagen/QSBlockTagsProvider.java b/src/main/java/dev/wolfieboy09/qstorage/datagen/QSBlockTagsProvider.java
index f868dcb..521a814 100644
--- a/src/main/java/dev/wolfieboy09/qstorage/datagen/QSBlockTagsProvider.java
+++ b/src/main/java/dev/wolfieboy09/qstorage/datagen/QSBlockTagsProvider.java
@@ -18,7 +18,8 @@ public QSBlockTagsProvider(PackOutput output, CompletableFuture<HolderLookup.Pro
 
     @Override
     public void addTags(@NotNull HolderLookup.Provider provider) {
-        tag(Tags.Blocks.NEEDS_GOLD_TOOL)
-                .add(QSBlocks.DISK_ASSEMBLER.get());
+        tag(Tags.Blocks.NEEDS_WOOD_TOOL)
+                .add(QSBlocks.DISK_ASSEMBLER.get())
+                .add(QSBlocks.STORAGE_MATRIX.get());
     }
 }