diff --git a/06_assistants_code_interpreter/assistants_code_interpreter.ipynb b/06_assistants_code_interpreter/assistants_code_interpreter.ipynb
index 171c00e0..b3c3d3ed 100644
--- a/06_assistants_code_interpreter/assistants_code_interpreter.ipynb
+++ b/06_assistants_code_interpreter/assistants_code_interpreter.ipynb
@@ -125,7 +125,7 @@
    "source": [
     "from openai.types.beta.threads.thread_message import ThreadMessage\n",
     "\n",
-    "message = client.beta.threads.messages.create(\n",
+    "message: ThreadMessage = client.beta.threads.messages.create(\n",
     "    thread_id=thread.id,\n",
     "    role=\"user\",\n",
     "    content=\"I need to solve the equation `3x + 11 = 14`. Can you help me?\"\n",
diff --git a/07_assistants_knowledge_retrieval/assistants_knowledge_retrieval.ipynb b/07_assistants_knowledge_retrieval/assistants_knowledge_retrieval.ipynb
index 9a17e7ad..0d577157 100644
--- a/07_assistants_knowledge_retrieval/assistants_knowledge_retrieval.ipynb
+++ b/07_assistants_knowledge_retrieval/assistants_knowledge_retrieval.ipynb
@@ -179,7 +179,7 @@
    "source": [
     "from openai.types.beta.threads.thread_message import ThreadMessage\n",
     "\n",
-    "message = client.beta.threads.messages.create(\n",
+    "message: ThreadMessage = client.beta.threads.messages.create(\n",
     "    thread_id=thread.id,\n",
     "    role=\"user\",\n",
     "    content=\"When and which city Zia U. Khan was born?\"\n",
diff --git a/08_assistants_function_calling/assistants_function_calling.ipynb b/08_assistants_function_calling/assistants_function_calling.ipynb
index 69c84599..156bb05f 100644
--- a/08_assistants_function_calling/assistants_function_calling.ipynb
+++ b/08_assistants_function_calling/assistants_function_calling.ipynb
@@ -207,7 +207,7 @@
     "from openai.types.beta.threads.thread_message import ThreadMessage\n",
     "\n",
     "# First Request\n",
-    "message = client.beta.threads.messages.create(\n",
+    "message: ThreadMessage = client.beta.threads.messages.create(\n",
     "    thread_id=thread.id,\n",
     "    role=\"user\",\n",
     "    content=\"How is the weather in Los Angles?\"\n",