diff --git a/examples/query_planner_execution/query_planner_execution.py b/examples/query_planner_execution/query_planner_execution.py
index 9a8eb50..b2c3d1a 100644
--- a/examples/query_planner_execution/query_planner_execution.py
+++ b/examples/query_planner_execution/query_planner_execution.py
@@ -138,7 +138,7 @@ def query_planner(question: str, plan=False) -> QueryPlan:
             max_tokens=1000,
         )
 
-        messages.append(completion.choices[0].message)
+        messages.append(completion['choices'][0]['message'])
 
         messages.append(
             {
diff --git a/instructor/function_calls.py b/instructor/function_calls.py
index e15a498..50d9249 100644
--- a/instructor/function_calls.py
+++ b/instructor/function_calls.py
@@ -101,7 +101,7 @@ class openai_function:
         Returns:
             result (any): result of the function call
         """
-        message = completion.choices[0].message
+        message = completion['choices'][0]['message']
 
         if throw_error:
             assert "function_call" in message, "No function call detected"
@@ -159,7 +159,7 @@ class OpenAISchema(BaseModel):
         Returns:
             cls (OpenAISchema): An instance of the class
         """
-        message = completion.choices[0].message
+        message = completion['choices'][0]['message']
 
         if throw_error:
             assert "function_call" in message, "No function call detected"