diff --git a/02_LangChain_Inputs_and_Outputs/code.ipynb b/02_LangChain_Inputs_and_Outputs/code.ipynb
index 20aa8088..e29d0b9e 100644
--- a/02_LangChain_Inputs_and_Outputs/code.ipynb
+++ b/02_LangChain_Inputs_and_Outputs/code.ipynb
@@ -115,7 +115,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.schema import SystemMessage, HumanMessage, AIMessage\n",
+    "from langchain_core.messages import SystemMessage, HumanMessage, AIMessage\n",
     "\n",
     "messages = [\n",
     "    SystemMessage(content=\"You are a helpful assistant specialized in providing information about BellaVista Italian Restaurant.\"),\n",
@@ -131,7 +131,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "llm_result = llm.invoke(messages=messages)\n",
+    "llm_result = llm.invoke(messages)\n",
     "llm_result"
    ]
   },