diff --git a/tests/Services/UpgradeManagerTest.php b/tests/Services/UpgradeManagerTest.php
index e319530..f55e8fe 100644
--- a/tests/Services/UpgradeManagerTest.php
+++ b/tests/Services/UpgradeManagerTest.php
@@ -112,7 +112,7 @@ public function testRunUpdatesVersionNoAssetCompile(): void
 
         $manager->run(new RunConfiguration(true, true), $this->createMock(OutputInterface::class));
 
-        static::assertCount(2, $consoleCommands);
+        static::assertCount(4, $consoleCommands);
         static::assertSame(['system:update:finish', '--skip-asset-build'], $consoleCommands[0]);
     }
 
@@ -150,7 +150,7 @@ public function testRunWithDifferentSalesChannelUrl(): void
 
         $manager->run(new RunConfiguration(), $this->createMock(OutputInterface::class));
 
-        static::assertCount(3, $consoleCommands);
+        static::assertCount(5, $consoleCommands);
         static::assertSame(['sales-channel:create:storefront', '--name=Storefront', '--url=http://localhost'], $consoleCommands[0]);
     }
 }