diff --git a/docs/docs/integrations/text_embedding/google_vertex_ai_palm.ipynb b/docs/docs/integrations/text_embedding/google_vertex_ai_palm.ipynb
index 4f07b8245d2..5bffb6e2fc9 100644
--- a/docs/docs/integrations/text_embedding/google_vertex_ai_palm.ipynb
+++ b/docs/docs/integrations/text_embedding/google_vertex_ai_palm.ipynb
@@ -103,7 +103,9 @@
       "cell_type": "markdown",
       "id": "c84fb993",
       "metadata": {},
-      "source": "To enable automated tracing of your model calls, set your [LangSmith](https://docs.smith.langchain.com/) API key:"
+      "source": [
+        "To enable automated tracing of your model calls, set your [LangSmith](https://docs.smith.langchain.com/) API key:"
+      ]
     },
     {
       "cell_type": "code",
@@ -157,7 +159,7 @@
         "from langchain_google_vertexai import VertexAIEmbeddings\n",
         "\n",
         "# Initialize the a specific Embeddings Model version\n",
-        "embeddings = VertexAIEmbeddings(model_name=\"text-embedding-004\")"
+        "embeddings = VertexAIEmbeddings(model_name=\"gemini-embedding-001\")"
       ]
     },
     {