forgot mm check

GregTechCEu · Dec 12, 2024 · 949e5a2 · 949e5a2
1 parent 5e8f8d9
commit 949e5a2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/com/gregtechceu/gtceu/common/item/ColorSprayBehaviour.java b/src/main/java/com/gregtechceu/gtceu/common/item/ColorSprayBehaviour.java
@@ -458,10 +458,11 @@ private static boolean tryStripBlockColor(Player player, Level world, BlockPos p
 
         // MTE special case
         BlockEntity be = world.getBlockEntity(pos);
-        if (be instanceof IMachineBlockEntity machineBe) {
+        if (be instanceof IMachineBlockEntity machineBe && first instanceof IMachineBlockEntity og) {
             MetaMachine mte = machineBe.getMetaMachine();
-            if (mte != null) {
-                if (mte.isPainted()) {
+            MetaMachine ogMte = og.getMetaMachine();
+            if (mte != null && ogMte != null) {
+                if (mte.isPainted() && mte.getPaintingColor() == ogMte.getPaintingColor()) {
                     mte.setPaintingColor(mte.getDefaultPaintingColor());
                     return true;
                 } else return false;