diff --git a/src/main/java/com/gtnewhorizon/structurelib/structure/StructureDefinition.java b/src/main/java/com/gtnewhorizon/structurelib/structure/StructureDefinition.java
index 5b1edc48..69ab1ea2 100644
--- a/src/main/java/com/gtnewhorizon/structurelib/structure/StructureDefinition.java
+++ b/src/main/java/com/gtnewhorizon/structurelib/structure/StructureDefinition.java
@@ -164,7 +164,7 @@ public class StructureDefinition<T> implements IStructureDefinition<T> {
                     socketLocations);
         }
 
-        private void compileShape(String name, String[][] structurePiece) {
+        private Builder<T> compileShape(String name, String[][] structurePiece) {
             StringBuilder builder = new StringBuilder();
 
             if (structurePiece.length > 0) {