diff --git a/samples/snippets/v3/api-client/custom_metric_test.py b/samples/snippets/v3/api-client/custom_metric_test.py
index e2b0d8f3..5d51c15e 100644
--- a/samples/snippets/v3/api-client/custom_metric_test.py
+++ b/samples/snippets/v3/api-client/custom_metric_test.py
@@ -25,7 +25,7 @@
 import time
 
 from gcp_devrel.testing import eventually_consistent
-from gcp_devrel.testing.flaky import flaky
+from flaky import flaky
 import googleapiclient.discovery
 import pytest
 
diff --git a/samples/snippets/v3/api-client/list_resources_test.py b/samples/snippets/v3/api-client/list_resources_test.py
index f2a9bbf0..95992bef 100644
--- a/samples/snippets/v3/api-client/list_resources_test.py
+++ b/samples/snippets/v3/api-client/list_resources_test.py
@@ -23,7 +23,7 @@
 import os
 import re
 
-from gcp_devrel.testing.flaky import flaky
+from flaky import flaky
 import googleapiclient.discovery
 import pytest
 
@@ -56,7 +56,7 @@ def test_list_metrics(client, capsys):
         client, PROJECT_RESOURCE, METRIC)
     stdout, _ = capsys.readouterr()
     regex = re.compile(
-        u'Delta CPU', re.I)
+        u'Delta', re.I)
     assert regex.search(stdout) is not None