diff --git a/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TESocket.java b/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TESocket.java
index b8595702..b4047d94 100644
--- a/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TESocket.java
+++ b/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TESocket.java
@@ -1,5 +1,6 @@
 package WayofTime.alchemicalWizardry.common.tileEntity;
 
+import WayofTime.alchemicalWizardry.api.items.interfaces.ArmourUpgrade;
 import WayofTime.alchemicalWizardry.common.NewPacketHandler;
 import net.minecraft.entity.player.EntityPlayer;
 import net.minecraft.inventory.IInventory;
@@ -132,6 +133,6 @@ public class TESocket extends TEInventory
     @Override
     public boolean isItemValidForSlot(int slot, ItemStack itemstack)
     {
-        return slot == 0;
+        return itemstack.getItem() instanceof ArmourUpgrade;
     }
 }
diff --git a/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TETeleposer.java b/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TETeleposer.java
index c07fb1af..7e35cb44 100644
--- a/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TETeleposer.java
+++ b/src/main/java/WayofTime/alchemicalWizardry/common/tileEntity/TETeleposer.java
@@ -246,7 +246,7 @@ public class TETeleposer extends TEInventory
     @Override
     public boolean isItemValidForSlot(int slot, ItemStack itemstack)
     {
-        return slot == 0;
+        return itemstack.getItem() instanceof TelepositionFocus;
     }
 
 }