diff --git a/google/cloud/aiplatform/utils/pipeline_utils.py b/google/cloud/aiplatform/utils/pipeline_utils.py
index 506dbdff26..5a09a98f18 100644
--- a/google/cloud/aiplatform/utils/pipeline_utils.py
+++ b/google/cloud/aiplatform/utils/pipeline_utils.py
@@ -138,7 +138,7 @@ def build(self) -> Dict[str, Any]:
 
     def _get_vertex_value(
         self, name: str, value: Union[int, float, str, bool, list, dict]
-    ) -> Union[Dict[str, Any], int, float, str, bool, list, dict]:
+    ) -> Union[int, float, str, bool, list, dict]:
         """Converts primitive values into Vertex pipeline Value proto message.
 
         Args: