diff --git a/src/main/java/org/embeddedt/vintagefix/mixin/dynamic_resources/MixinBlockModelShapes.java b/src/main/java/org/embeddedt/vintagefix/mixin/dynamic_resources/MixinBlockModelShapes.java
index 3f626db..b3a3e76 100644
--- a/src/main/java/org/embeddedt/vintagefix/mixin/dynamic_resources/MixinBlockModelShapes.java
+++ b/src/main/java/org/embeddedt/vintagefix/mixin/dynamic_resources/MixinBlockModelShapes.java
@@ -67,16 +67,20 @@ private IBakedModel getModelForStateSlow(IBlockState state) {
      **/
     @Overwrite
     public IBakedModel getModelForState(IBlockState state) {
-        IModelHoldingBlockState holder = (IModelHoldingBlockState)state;
-        IBakedModel model = holder.vfix$getModel();
+        if(state instanceof IModelHoldingBlockState) {
+            IModelHoldingBlockState holder = (IModelHoldingBlockState)state;
+            IBakedModel model = holder.vfix$getModel();
 
-        if(model != null) {
+            if(model != null) {
+                return model;
+            }
+
+            model = this.getModelForStateSlow(state);
+            holder.vfix$setModel(model);
             return model;
+        } else {
+            return this.getModelForStateSlow(state);
         }
-
-        model = this.getModelForStateSlow(state);
-        holder.vfix$setModel(model);
-        return model;
     }
 
     @Override