diff --git a/streamlit_app/visualizer/api.py b/streamlit_app/visualizer/api.py
index a2104e9a..328d33ef 100644
--- a/streamlit_app/visualizer/api.py
+++ b/streamlit_app/visualizer/api.py
@@ -63,7 +63,7 @@ class SkyvernClient:
         response = requests.get(url, params=params, headers=headers)
         steps = response.json()
         for step in steps:
-            if step.output and "actions_and_results" in step.output:
+            if "output" in step and "actions_and_results" in step["output"]:
                 step["output"]["actions_and_results"] = json.dumps(step["output"]["actions_and_results"])
         return steps