diff --git a/samples/snippets/product_search/import_product_sets.py b/samples/snippets/product_search/import_product_sets.py
index d08c15c5..e2d03714 100755
--- a/samples/snippets/product_search/import_product_sets.py
+++ b/samples/snippets/product_search/import_product_sets.py
@@ -24,7 +24,7 @@
 import argparse
 
 # [START vision_product_search_tutorial_import]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 # [END vision_product_search_tutorial_import]
 
 
diff --git a/samples/snippets/product_search/product_in_product_set_management.py b/samples/snippets/product_search/product_in_product_set_management.py
index e9bfad9b..8c08c8a0 100755
--- a/samples/snippets/product_search/product_in_product_set_management.py
+++ b/samples/snippets/product_search/product_in_product_set_management.py
@@ -25,7 +25,7 @@
 
 # [START vision_product_search_add_product_to_product_set]
 # [START vision_product_search_remove_product_from_product_set]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_add_product_to_product_set]
 # [END vision_product_search_remove_product_from_product_set]
diff --git a/samples/snippets/product_search/product_in_product_set_management_test.py b/samples/snippets/product_search/product_in_product_set_management_test.py
index cae2d2ca..21a23bdb 100644
--- a/samples/snippets/product_search/product_in_product_set_management_test.py
+++ b/samples/snippets/product_search/product_in_product_set_management_test.py
@@ -54,13 +54,13 @@ def product_and_product_set():
 def test_add_product_to_product_set(capsys, product_and_product_set):
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID not in out
+    assert 'Product id: {}'.format(PRODUCT_ID) not in out
 
     add_product_to_product_set(
         PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID)
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID in out
+    assert 'Product id: {}'.format(PRODUCT_ID) in out
 
 
 def test_remove_product_from_product_set(capsys, product_and_product_set):
@@ -68,10 +68,10 @@ def test_remove_product_from_product_set(capsys, product_and_product_set):
         PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID)
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID in out
+    assert 'Product id: {}'.format(PRODUCT_ID) in out
 
     remove_product_from_product_set(
         PROJECT_ID, LOCATION, PRODUCT_ID, PRODUCT_SET_ID)
     list_products_in_product_set(PROJECT_ID, LOCATION, PRODUCT_SET_ID)
     out, _ = capsys.readouterr()
-    assert PRODUCT_ID not in out
+    assert 'Product id: {}'.format(PRODUCT_ID) not in out
diff --git a/samples/snippets/product_search/product_management.py b/samples/snippets/product_search/product_management.py
index 4b187e98..01c05714 100755
--- a/samples/snippets/product_search/product_management.py
+++ b/samples/snippets/product_search/product_management.py
@@ -28,7 +28,7 @@
 # [START vision_product_search_list_products]
 # [START vision_product_search_get_product]
 # [START vision_product_search_update_product_labels]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_create_product]
 # [END vision_product_search_delete_product]
diff --git a/samples/snippets/product_search/product_search.py b/samples/snippets/product_search/product_search.py
index 05f19ea3..343bb6d0 100755
--- a/samples/snippets/product_search/product_search.py
+++ b/samples/snippets/product_search/product_search.py
@@ -26,7 +26,7 @@
 
 # [START vision_product_search_get_similar_products]
 # [START vision_product_search_get_similar_products_gcs]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_get_similar_products]
 # [END vision_product_search_get_similar_products_gcs]
diff --git a/samples/snippets/product_search/product_set_management.py b/samples/snippets/product_search/product_set_management.py
index 941f59fa..7964bc27 100755
--- a/samples/snippets/product_search/product_set_management.py
+++ b/samples/snippets/product_search/product_set_management.py
@@ -27,7 +27,7 @@
 # [START vision_product_search_list_product_sets]
 # [START vision_product_search_get_product_set]
 # [START vision_product_search_create_product_set]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_delete_product_set]
 # [END vision_product_search_list_product_sets]
diff --git a/samples/snippets/product_search/reference_image_management.py b/samples/snippets/product_search/reference_image_management.py
index a5392783..7e546b7e 100755
--- a/samples/snippets/product_search/reference_image_management.py
+++ b/samples/snippets/product_search/reference_image_management.py
@@ -27,7 +27,7 @@
 # [START vision_product_search_delete_reference_image]
 # [START vision_product_search_list_reference_images]
 # [START vision_product_search_get_reference_image]
-from google.cloud import vision_v1p3beta1 as vision
+from google.cloud import vision
 
 # [END vision_product_search_create_reference_image]
 # [END vision_product_search_delete_reference_image]
diff --git a/samples/snippets/product_search/requirements.txt b/samples/snippets/product_search/requirements.txt
index 01a92ffc..e0dc4537 100644
--- a/samples/snippets/product_search/requirements.txt
+++ b/samples/snippets/product_search/requirements.txt
@@ -1 +1 @@
-google-cloud-vision==0.35.0
+google-cloud-vision==0.35.1