diff --git a/src/main/java/ru/magistu/siegemachines/plugins/jei/SiegeWorkbenchRecipeCategory.java b/src/main/java/ru/magistu/siegemachines/plugins/jei/SiegeWorkbenchRecipeCategory.java
index 997947a..5f3f3b8 100644
--- a/src/main/java/ru/magistu/siegemachines/plugins/jei/SiegeWorkbenchRecipeCategory.java
+++ b/src/main/java/ru/magistu/siegemachines/plugins/jei/SiegeWorkbenchRecipeCategory.java
@@ -42,10 +42,25 @@ public class SiegeWorkbenchRecipeCategory implements IRecipeCategory<SiegeWorkbe
         return Component.translatable("category." + SiegeMachines.ID + ".siege_workbench_crafting");
     }
 
-//    @Override
-//    public IDrawable getBackground() {
-//        return this.background;
-//    }
+    @Override
+    public int getWidth() {
+        IDrawable background = this.background;
+        if (background == null) {
+            throw new IllegalStateException("getWidth() and getHeight() must be overridden if background is null");
+        } else {
+            return background.getWidth();
+        }
+    }
+
+    @Override
+    public int getHeight() {
+        IDrawable background = this.background;
+        if (background == null) {
+            throw new IllegalStateException("getWidth() and getHeight() must be overridden if background is null");
+        } else {
+            return background.getHeight();
+        }
+    }
 
     @Override
     public IDrawable getIcon() {