diff --git a/client/gui/AEBaseGui.java b/client/gui/AEBaseGui.java
index 17971c1ab..5e7fbbfa5 100644
--- a/client/gui/AEBaseGui.java
+++ b/client/gui/AEBaseGui.java
@@ -42,7 +42,6 @@ import appeng.core.AELog;
 import appeng.core.sync.network.NetworkHandler;
 import appeng.core.sync.packets.PacketInventoryAction;
 import appeng.helpers.InventoryAction;
-import appeng.parts.automation.UpgradeInventory;
 import appeng.util.Platform;
 
 public abstract class AEBaseGui extends GuiContainer
@@ -560,7 +559,7 @@ public abstract class AEBaseGui extends GuiContainer
 					if ( ((AppEngSlot) s).isValid == hasCalculatedValidness.NotAvailable )
 					{
 						boolean isValid = s.isItemValid( is ) || s instanceof SlotOutput || s instanceof AppEngCraftingSlot || s instanceof SlotDisabled
-								|| s instanceof SlotInaccessable || s instanceof SlotFake || s.inventory instanceof UpgradeInventory;
+								|| s instanceof SlotInaccessable || s instanceof SlotFake || s instanceof SlotRestrictedInput;
 						if ( isValid && s instanceof SlotRestrictedInput )
 						{
 							try