diff --git a/tests/machine_learning/20_trained_model_serverless.yml b/tests/machine_learning/20_trained_model_serverless.yml
index 4f22fe8..27f48a3 100644
--- a/tests/machine_learning/20_trained_model_serverless.yml
+++ b/tests/machine_learning/20_trained_model_serverless.yml
@@ -52,7 +52,8 @@ teardown:
       ml.start_trained_model_deployment:
         model_id: test_model
         priority: "low"
-        wait_for: started
+        wait_for: fully_allocated
+        timeout: "60s"
   - match: { assignment.task_parameters.model_id: test_model }
   - do:
       ml.update_trained_model_deployment: