diff --git a/invui/src/main/java/xyz/xenondevs/invui/gui/AbstractGui.java b/invui/src/main/java/xyz/xenondevs/invui/gui/AbstractGui.java
index e1d9dff..5bf612a 100644
--- a/invui/src/main/java/xyz/xenondevs/invui/gui/AbstractGui.java
+++ b/invui/src/main/java/xyz/xenondevs/invui/gui/AbstractGui.java
@@ -451,6 +451,7 @@ public abstract class AbstractGui implements Gui, GuiParent {
         }
     }
     
+    @SuppressWarnings("unchecked")
     @Override
     public void setSlotElement(int index, SlotElement slotElement) {
         SlotElement oldElement = slotElements[index];
@@ -462,7 +463,7 @@ public abstract class AbstractGui implements Gui, GuiParent {
         if (slotElement instanceof SlotElement.ItemSlotElement) {
             Item item = ((SlotElement.ItemSlotElement) slotElement).getItem();
             if (item instanceof ControlItem<?>)
-                ((ControlItem<?>) item).setGui(this);
+                ((ControlItem<Gui>) item).setGui(this);
         }
         
         // notify parents that a SlotElement has been changed
diff --git a/invui/src/main/java/xyz/xenondevs/invui/item/impl/controlitem/ControlItem.java b/invui/src/main/java/xyz/xenondevs/invui/item/impl/controlitem/ControlItem.java
index de3f35d..11c5d6b 100644
--- a/invui/src/main/java/xyz/xenondevs/invui/item/impl/controlitem/ControlItem.java
+++ b/invui/src/main/java/xyz/xenondevs/invui/item/impl/controlitem/ControlItem.java
@@ -25,9 +25,10 @@ public abstract class ControlItem<G extends Gui> extends AbstractItem {
         return gui;
     }
     
-    @SuppressWarnings("unchecked")
-    public void setGui(Object gui) {
-        if (this.gui == null) this.gui = (G) gui;
+    public void setGui(G gui) {
+        if (this.gui == null) {
+            this.gui = gui;
+        }
     }
     
 }