From 39993a176ffb3350f7b2a559e0821b59664c15eb Mon Sep 17 00:00:00 2001
From: psikomonkie <189469115+psikomonkie@users.noreply.github.com>
Date: Wed, 8 Jan 2025 16:46:29 -0500
Subject: [PATCH] Issue 3650: Fixed Restore Unit Action Test

---
 .../mekhq/campaign/unit/actions/RestoreUnitActionTest.java     | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java b/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java
index 174fc901f2..d2312965c2 100644
--- a/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java
+++ b/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java
@@ -59,8 +59,6 @@ public void restoreUnitSwitchesOutEntityAndParts() {
         Unit unit = mock(Unit.class);
         when(unit.getId()).thenReturn(id);
         when(unit.getEntity()).thenReturn(mockEntity);
-        doNothing().when(unit).initializeShipTransportSpace();
-        doNothing().when(unit).initializeTacticalTransportSpace();
 
         Entity mockNewEntity = mock(Entity.class);
         doAnswer(inv -> {
@@ -86,6 +84,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)).initializeParts(eq(true));
         verify(unit, times(1)).runDiagnostic(eq(false));
         verify(unit, times(1)).setSalvage(eq(false));