diff --git a/vertexai/_genai/datasets.py b/vertexai/_genai/datasets.py
index 0d9900e47e..2b6b71334e 100644
--- a/vertexai/_genai/datasets.py
+++ b/vertexai/_genai/datasets.py
@@ -1085,7 +1085,7 @@ def update_multimodal_dataset(
         self,
         *,
         multimodal_dataset: types.MultimodalDatasetOrDict,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Updates a multimodal dataset.
 
@@ -1109,9 +1109,9 @@ def update_multimodal_dataset(
         _datasets_utils.validate_multimodal_dataset_bigquery_uri(multimodal_dataset)
 
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.UpdateMultimodalDatasetConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.UpdateMultimodalDatasetConfig()
 
         return self._update_multimodal_dataset(
             config=config,
@@ -1125,7 +1125,7 @@ def get_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Gets a multimodal dataset.
 
@@ -1141,9 +1141,9 @@ def get_multimodal_dataset(
           dataset.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return self._get_multimodal_dataset(config=config, name=name)
 
@@ -1151,7 +1151,7 @@ def delete_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDatasetOperation:
         """Deletes a multimodal dataset.
 
@@ -1167,9 +1167,9 @@ def delete_multimodal_dataset(
           multimodal dataset operation.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return self._delete_multimodal_dataset(config=config, name=name)
 
@@ -2309,7 +2309,7 @@ async def update_multimodal_dataset(
         self,
         *,
         multimodal_dataset: types.MultimodalDatasetOrDict,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.UpdateMultimodalDatasetConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Updates a multimodal dataset.
 
@@ -2329,9 +2329,9 @@ async def update_multimodal_dataset(
         _datasets_utils.validate_multimodal_dataset_bigquery_uri(multimodal_dataset)
 
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.UpdateMultimodalDatasetConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.UpdateMultimodalDatasetConfig()
 
         return await self._update_multimodal_dataset(
             config=config,
@@ -2345,7 +2345,7 @@ async def get_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDataset:
         """Gets a multimodal dataset.
 
@@ -2361,9 +2361,9 @@ async def get_multimodal_dataset(
           dataset.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return await self._get_multimodal_dataset(config=config, name=name)
 
@@ -2371,7 +2371,7 @@ async def delete_multimodal_dataset(
         self,
         *,
         name: str,
-        config: Optional[types.CreateMultimodalDatasetConfigOrDict] = None,
+        config: Optional[types.VertexBaseConfigOrDict] = None,
     ) -> types.MultimodalDatasetOperation:
         """Deletes a multimodal dataset.
 
@@ -2387,9 +2387,9 @@ async def delete_multimodal_dataset(
           multimodal dataset operation.
         """
         if isinstance(config, dict):
-            config = types.CreateMultimodalDatasetConfig(**config)
+            config = types.VertexBaseConfig(**config)
         elif not config:
-            config = types.CreateMultimodalDatasetConfig()
+            config = types.VertexBaseConfig()
 
         return await self._delete_multimodal_dataset(config=config, name=name)