diff --git a/src/main/java/com/github/minecraftschurlimods/bibliocraft/client/model/TableModel.java b/src/main/java/com/github/minecraftschurlimods/bibliocraft/client/model/TableModel.java
index 763d191a..f5ddac08 100644
--- a/src/main/java/com/github/minecraftschurlimods/bibliocraft/client/model/TableModel.java
+++ b/src/main/java/com/github/minecraftschurlimods/bibliocraft/client/model/TableModel.java
@@ -1,7 +1,6 @@
 package com.github.minecraftschurlimods.bibliocraft.client.model;
 
 import com.github.minecraftschurlimods.bibliocraft.content.table.TableBlock;
-import com.github.minecraftschurlimods.bibliocraft.content.table.TableBlockEntity;
 import com.github.minecraftschurlimods.bibliocraft.util.BCUtil;
 import com.google.gson.JsonObject;
 import com.mojang.blaze3d.vertex.PoseStack;
@@ -31,7 +30,6 @@
 import net.neoforged.neoforge.common.data.ExistingFileHelper;
 import org.jetbrains.annotations.Nullable;
 
-import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -54,8 +52,11 @@ public TableModel(boolean useAmbientOcclusion, boolean isGui3d, boolean usesBloc
 
     @Override
     public List<BakedQuad> getQuads(@Nullable BlockState state, @Nullable Direction side, RandomSource rand, ModelData extraData, @Nullable RenderType renderType) {
-        TableBlock.Type type = extraData.get(TableBlockEntity.TYPE_PROPERTY);
-        return new ArrayList<>(baseMap.get(type).getQuads(state, side, rand, extraData, renderType));
+        TableBlock.Type type = TableBlock.Type.NONE;
+        if (state != null && state.hasProperty(TableBlock.TYPE)) {
+            type = state.getValue(TableBlock.TYPE);
+        }
+        return baseMap.get(type).getQuads(state, side, rand, extraData, renderType);
     }
 
     @Override
diff --git a/src/main/java/com/github/minecraftschurlimods/bibliocraft/content/table/TableBlockEntity.java b/src/main/java/com/github/minecraftschurlimods/bibliocraft/content/table/TableBlockEntity.java
index 457977d6..00c67e5c 100644
--- a/src/main/java/com/github/minecraftschurlimods/bibliocraft/content/table/TableBlockEntity.java
+++ b/src/main/java/com/github/minecraftschurlimods/bibliocraft/content/table/TableBlockEntity.java
@@ -10,14 +10,10 @@
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.state.BlockState;
-import net.neoforged.neoforge.client.model.data.ModelData;
-import net.neoforged.neoforge.client.model.data.ModelProperty;
 
 import java.util.Objects;
 
 public class TableBlockEntity extends BCBlockEntity {
-    public static final ModelProperty<TableBlock.Type> TYPE_PROPERTY = new ModelProperty<>();
-
     public TableBlockEntity(BlockPos pos, BlockState state) {
         super(BCBlockEntities.TABLE.get(), 2, pos, state);
     }
@@ -42,9 +38,4 @@ public void setItem(int slot, ItemStack stack) {
             requestModelDataUpdate();
         }
     }
-
-    @Override
-    public ModelData getModelData() {
-        return ModelData.builder().with(TYPE_PROPERTY, getBlockState().getValue(TableBlock.TYPE)).build();
-    }
 }