Making SheetsMixin more extensible

1 files changed, 3 insertions, 7 deletions

diff --git a/src/main/java/lv/enes/mc/eris_alchemy/mixin/client/SheetsMixin.java b/src/main/java/lv/enes/mc/eris_alchemy/mixin/client/SheetsMixin.java
index 145a68c..bc9e7b8 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/mixin/client/SheetsMixin.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/mixin/client/SheetsMixin.java
@@ -1,8 +1,6 @@
 package lv.enes.mc.eris_alchemy.mixin.client;
 
-import lv.enes.mc.eris_alchemy.block.entity.EnergyCondenserEntity;
-import lv.enes.mc.eris_alchemy.client.ErisAlchemyMaterials;
-import lv.enes.mc.eris_alchemy.block.entity.AlchemicalChestEntity;
+import lv.enes.mc.eris_alchemy.block.entity.ChestLikeEntity;
 import net.minecraft.client.renderer.Sheets;
 import net.minecraft.client.resources.model.Material;
 import net.minecraft.world.level.block.entity.BlockEntity;
@@ -25,10 +23,8 @@ public abstract class SheetsMixin {
                         boolean christmas,
                         CallbackInfoReturnable<Material> cir
         ) {
-                if (entity instanceof AlchemicalChestEntity) {
-                        cir.setReturnValue(ErisAlchemyMaterials.ALCHEMICAL_CHEST);
-                } else if (entity instanceof EnergyCondenserEntity) {
-                        cir.setReturnValue(ErisAlchemyMaterials.ENERGY_CONDENSER);
+                if (entity instanceof ChestLikeEntity chestlike) {
+                        cir.setReturnValue(chestlike.material);
                 }
         }
 }