diff --git a/gemini/batch-prediction/intro_batch_prediction.ipynb b/gemini/batch-prediction/intro_batch_prediction.ipynb
index dbfe70d1b0e..84bd972d812 100644
--- a/gemini/batch-prediction/intro_batch_prediction.ipynb
+++ b/gemini/batch-prediction/intro_batch_prediction.ipynb
@@ -253,7 +253,7 @@
         "\n",
         "from google.cloud import storage\n",
         "from vertexai.generative_models import GenerativeModel\n",
-        "from vertexai.preview.batch_prediction import BatchPredictionJob"
+        "from vertexai.batch_prediction import BatchPredictionJob"
       ]
     },
     {
diff --git a/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb b/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb
index 8f082edbe3e..e83d0945994 100644
--- a/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb
+++ b/gemini/batch-prediction/intro_batch_prediction_using_bigquery_input.ipynb
@@ -247,8 +247,8 @@
         "\n",
         "from google.cloud import bigquery\n",
         "import vertexai\n",
-        "from vertexai.generative_models import GenerativeModel\n",
-        "from vertexai.preview.batch_prediction import BatchPredictionJob"
+        "from vertexai.batch_prediction import BatchPredictionJob\n",
+        "from vertexai.generative_models import GenerativeModel"
       ]
     },
     {