diff --git a/src/Sylius/Behat/Context/Cli/InstallerContext.php b/src/Sylius/Behat/Context/Cli/InstallerContext.php
index 7913682243..035a7607a1 100644
--- a/src/Sylius/Behat/Context/Cli/InstallerContext.php
+++ b/src/Sylius/Behat/Context/Cli/InstallerContext.php
@@ -70,7 +70,7 @@ final class InstallerContext implements Context
     public function iRunSyliusCommandLineInstaller(): void
     {
         $this->application = new Application($this->kernel);
-        $this->application->addCommand(new SetupCommand(
+        $this->addCommandHelper(new SetupCommand(
             $this->entityManager,
             $this->commandDirectoryChecker,
             $this->currencySetup,
@@ -91,7 +91,7 @@ final class InstallerContext implements Context
     public function iRunSyliusInstallSampleDataCommand(): void
     {
         $this->application = new Application($this->kernel);
-        $this->application->addCommand(new InstallSampleDataCommand(
+        $this->addCommandHelper(new InstallSampleDataCommand(
             $this->entityManager,
             $this->commandDirectoryChecker,
             $this->publicDir,
@@ -158,4 +158,15 @@ final class InstallerContext implements Context
         } catch (\Exception) {
         }
     }
+
+    private function addCommandHelper(Command $command): void
+    {
+        if (method_exists($this->application, 'addCommand')) {
+            $this->application->addCommand($command);
+
+            return;
+        }
+
+        $this->application->add($command);
+    }
 }