diff --git a/src/main/java/dev/wolfieboy09/qstorage/block/disk_assembler/DiskAssemblerBlock.java b/src/main/java/dev/wolfieboy09/qstorage/block/disk_assembler/DiskAssemblerBlock.java
index ec74db4..47539c8 100644
--- a/src/main/java/dev/wolfieboy09/qstorage/block/disk_assembler/DiskAssemblerBlock.java
+++ b/src/main/java/dev/wolfieboy09/qstorage/block/disk_assembler/DiskAssemblerBlock.java
@@ -2,6 +2,7 @@
 
 import com.mojang.serialization.MapCodec;
 import dev.wolfieboy09.qstorage.api.annotation.NothingNullByDefault;
+import dev.wolfieboy09.qstorage.block.AbstractBaseEntityBlock;
 import net.minecraft.core.BlockPos;
 import net.minecraft.server.level.ServerPlayer;
 import net.minecraft.world.*;
@@ -21,8 +22,7 @@
 import org.jetbrains.annotations.Nullable;
 
 @NothingNullByDefault
-@SuppressWarnings("deprecation")
-public class DiskAssemblerBlock extends BaseEntityBlock {
+public class DiskAssemblerBlock extends AbstractBaseEntityBlock {
     public static final DirectionProperty FACING = BlockStateProperties.HORIZONTAL_FACING;
     public static final MapCodec<DiskAssemblerBlock> CODEC = simpleCodec(DiskAssemblerBlock::new);
 
@@ -45,12 +45,6 @@ protected void createBlockStateDefinition(StateDefinition.Builder<Block, BlockSt
         return this.defaultBlockState().setValue(FACING, context.getHorizontalDirection().getOpposite());
     }
 
-    @Override
-    protected RenderShape getRenderShape(BlockState state) {
-        return RenderShape.MODEL;
-    }
-
-
     @Override
     public @Nullable BlockEntity newBlockEntity(BlockPos blockPos, BlockState blockState) {
         return new DiskAssemblerBlockEntity(blockPos, blockState);