diff --git a/tests/Feature/DisableCommandTest.php b/tests/Feature/DisableCommandTest.php
index e1d719e..ea91556 100644
--- a/tests/Feature/DisableCommandTest.php
+++ b/tests/Feature/DisableCommandTest.php
@@ -55,7 +55,7 @@ function it_can_disable_a_service_from_menu()
             array_push($disableableServices, '<info>Exit</>');
 
             $this->artisan('disable')
-                ->expectsChoice('Takeout containers to disable', $postgressName, $disableableServices)
+                ->expectsChoice('Takeout containers to disable', $postgressName, array_values($disableableServices))
                 ->assertExitCode(0);
         } else {
             $menuMock = $this->mock(Menu::class, function ($mock) use ($postgressId) {
diff --git a/tests/Feature/EnableCommandTest.php b/tests/Feature/EnableCommandTest.php
index 9ff0ea4..4f67b28 100644
--- a/tests/Feature/EnableCommandTest.php
+++ b/tests/Feature/EnableCommandTest.php
@@ -8,11 +8,6 @@
 use App\Services\MeiliSearch;
 use App\Services\PostgreSql;
 use App\Shell\Docker;
-use Illuminate\Console\Command;
-use Laravel\Prompts\Prompt;
-use Laravel\Prompts\SearchPrompt;
-use NunoMaduro\LaravelConsoleMenu\Menu;
-use PHPUnit\Framework\Assert;
 use Tests\TestCase;
 
 class EnableCommandTest extends TestCase
@@ -66,7 +61,7 @@ function it_can_enable_a_service_from_menu()
                 'Database: PostgreSQL',
                 'Search: MeiliSearch',
                 'meilisearch',
-                'postgresql',
+                $postgres,
             ];
 
             $this->artisan('enable')
@@ -75,11 +70,11 @@ function it_can_enable_a_service_from_menu()
 
             $menuItems = [
                 'Database: PostgreSQL',
-                'postgresql',
+                $postgres,
             ];
 
             $this->artisan('enable')
-                ->expectsChoice('Takeout containers to enable', 'PostgreSQL', $menuItems)
+                ->expectsChoice('Takeout containers to enable', $postgres, $menuItems)
                 ->assertExitCode(0);
         }
     }