diff --git a/tests/Feature/HetznerApiTest.php b/tests/Feature/HetznerApiTest.php
index 298475934..bd316ca49 100644
--- a/tests/Feature/HetznerApiTest.php
+++ b/tests/Feature/HetznerApiTest.php
@@ -16,7 +16,8 @@
     $this->team->members()->attach($this->user->id, ['role' => 'owner']);
 
     // Create an API token for the user
-    $this->token = $this->user->createToken('test-token', ['*'], $this->team->id);
+    session(['currentTeam' => $this->team]);
+    $this->token = $this->user->createToken('test-token', ['*']);
     $this->bearerToken = $this->token->plainTextToken;
 
     // Create a Hetzner cloud provider token