diff --git a/.azure-pipelines/test_plan.py b/.azure-pipelines/test_plan.py
index b759dfe3a82..7b85e090162 100644
--- a/.azure-pipelines/test_plan.py
+++ b/.azure-pipelines/test_plan.py
@@ -352,9 +352,11 @@ def poll(self, test_plan_id, interval=60, timeout=-1, expected_state=""):
                     steps = None
                     step_status = None
                     extra_params = resp_data.get("extra_params", None)
-
-                    if extra_params:
+                    runtime = resp_data.get("runtime", None)
+                    if extra_params and "steps" in extra_params:
                         steps = extra_params.get("steps", None)
+                    else:
+                        steps = runtime.get("steps", None)
                     if steps:
                         for step in steps:
                             if step.get("step") == expected_state: