Issue 3650: Fixed RestoreUnitActionTest.java

MegaMek · Jan 11, 2025 · 3443d6b · 3443d6b
1 parent 8db443e
commit 3443d6b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java b/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java
@@ -83,8 +83,7 @@ public void restoreUnitSwitchesOutEntityAndParts() {
 
         verify(unit, times(1)).setEntity(eq(mockNewEntity));
         verify(unit, times(1)).removeParts();
-        verify(unit, times(1)).initializeShipTransportSpace();
-        verify(unit, times(1)).initializeTacticalTransportSpace();
+        verify(unit, times(1)).initializeAllTransportSpace();
         verify(unit, times(1)).initializeParts(eq(true));
         verify(unit, times(1)).runDiagnostic(eq(false));
         verify(unit, times(1)).setSalvage(eq(false));