1 files changed, 7 insertions, 3 deletions
diff --git a/src/main/java/gregtech/api/multitileentity/multiblock/casing/UpgradeCasing.java b/src/main/java/gregtech/api/multitileentity/multiblock/casing/UpgradeCasing.java
index e3230eac02..fb045557e4 100644
--- a/src/main/java/gregtech/api/multitileentity/multiblock/casing/UpgradeCasing.java
+++ b/src/main/java/gregtech/api/multitileentity/multiblock/casing/UpgradeCasing.java
@@ -16,9 +16,12 @@ public abstract class UpgradeCasing extends MultiBlockPart {
     }
 
     @Override
-    public void setTarget(IMultiBlockController aTarget, int aAllowedModes) {
-        super.setTarget(aTarget, aAllowedModes);
-        if (target != null) customWork(target);
+    public void setTarget(IMultiBlockController newTarget, int aAllowedModes) {
+        super.setTarget(newTarget, aAllowedModes);
+
+        if (getTarget(false) != null) {
+            customWork(getTarget(false));
+        }
     }
 
     @Override
@@ -28,4 +31,5 @@ public abstract class UpgradeCasing extends MultiBlockPart {
     }
 
     protected abstract void customWork(IMultiBlockController aTarget);
+
 }