diff --git a/google/cloud/aiplatform/helpers/_decorators.py b/google/cloud/aiplatform/helpers/_decorators.py
index 5d9aa28beaa..95aac31c4fb 100644
--- a/google/cloud/aiplatform/helpers/_decorators.py
+++ b/google/cloud/aiplatform/helpers/_decorators.py
@@ -68,3 +68,5 @@ def _from_map(map_):
 
 marshal = Marshal(name="google.cloud.aiplatform.v1beta1")
 marshal.register(Value, ConversionValueRule(marshal=marshal))
+marshal = Marshal(name="google.cloud.aiplatform.v1")
+marshal.register(Value, ConversionValueRule(marshal=marshal))
diff --git a/samples/snippets/requirements.txt b/samples/snippets/requirements.txt
index b9fd33d5c1c..481213275fb 100644
--- a/samples/snippets/requirements.txt
+++ b/samples/snippets/requirements.txt
@@ -1,3 +1,3 @@
 pytest==6.2.2
 google-cloud-storage>=1.26.0, <2.0.0dev
-google-cloud-aiplatform==0.5.1
+google-cloud-aiplatform==0.6.0