diff --git a/tests/Unit/Domain/CombatEngineTest.php b/tests/Unit/Domain/CombatEngineTest.php
index d700481..90d2397 100644
--- a/tests/Unit/Domain/CombatEngineTest.php
+++ b/tests/Unit/Domain/CombatEngineTest.php
@@ -20,7 +20,7 @@ final class CombatEngineTest extends TestCase
         $moved = $unit->moveTo(new Position(1, 0))->spendAction();
         $next = $match->withUnit($moved);
 
-        self::assertSame('0:0', $unit->position->key());
+        self::assertSame('0:0', $match->unit('unit-1')->position->key());
         self::assertSame('1:0', $next->unit('unit-1')->position->key());
         self::assertSame(1, $next->unit('unit-1')->actionsRemaining);
     }