diff --git a/pipelines/examples/azure_openai_pipeline.py b/pipelines/examples/azure_openai_pipeline.py
index 2e1dcb6..0e8d77e 100644
--- a/pipelines/examples/azure_openai_pipeline.py
+++ b/pipelines/examples/azure_openai_pipeline.py
@@ -47,7 +47,7 @@ class Pipeline:
 
             r.raise_for_status()
 
-            if data["stream"]:
+            if body["stream"]:
                 return r.iter_lines()
             else:
                 return r.json()
diff --git a/pipelines/examples/openai_pipeline.py b/pipelines/examples/openai_pipeline.py
index 54f99c3..80b8439 100644
--- a/pipelines/examples/openai_pipeline.py
+++ b/pipelines/examples/openai_pipeline.py
@@ -44,7 +44,7 @@ class Pipeline:
 
             r.raise_for_status()
 
-            if data["stream"]:
+            if body["stream"]:
                 return r.iter_lines()
             else:
                 return r.json()