diff --git a/cloud-service-providers/google-cloud/vertexai/python/nim-vertexai.ipynb b/cloud-service-providers/google-cloud/vertexai/python/nim-vertexai.ipynb
index 0a149b85..576cd362 100644
--- a/cloud-service-providers/google-cloud/vertexai/python/nim-vertexai.ipynb
+++ b/cloud-service-providers/google-cloud/vertexai/python/nim-vertexai.ipynb
@@ -1178,6 +1178,7 @@
    },
    "outputs": [],
    "source": [
+    "import json\n",
     "messages = [\n",
     "    {\"role\": \"user\", \"content\": \"Hello! How are you?\"},\n",
     "    {\"role\": \"assistant\", \"content\": \"Hi! I am quite well, how can I help you today?\"},\n",