diff --git a/tests/store_model_in_db_tests/test_adding_passthrough_model.py b/tests/store_model_in_db_tests/test_adding_passthrough_model.py
index ad26e19bd6..e901be5bd7 100644
--- a/tests/store_model_in_db_tests/test_adding_passthrough_model.py
+++ b/tests/store_model_in_db_tests/test_adding_passthrough_model.py
@@ -142,7 +142,7 @@ def create_virtual_key():
         json={},
     )
     print(response.json())
-    return response.json()["token"]
+    return response.json()["key"]
 
 
 def add_assembly_ai_model_to_db(