diff --git a/optimization/snippets/async_api_test.py b/optimization/snippets/async_api_test.py
index a400d2c5aede..fce1b585f7cb 100644
--- a/optimization/snippets/async_api_test.py
+++ b/optimization/snippets/async_api_test.py
@@ -17,7 +17,8 @@
 import google.auth
 from google.cloud import storage
 import pytest
-from samples.snippets import async_api
+
+import async_api
 
 
 # TODO(developer): Replace the variables in the file before use.
diff --git a/optimization/snippets/noxfile.py b/optimization/snippets/noxfile.py
index a40410b56369..29b5bc852183 100644
--- a/optimization/snippets/noxfile.py
+++ b/optimization/snippets/noxfile.py
@@ -89,7 +89,7 @@ def get_pytest_env_vars() -> Dict[str, str]:
 
 # DO NOT EDIT - automatically generated.
 # All versions used to test samples.
-ALL_VERSIONS = ["3.6", "3.7", "3.8", "3.9", "3.10"]
+ALL_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
 
 # Any default versions that should be ignored.
 IGNORED_VERSIONS = TEST_CONFIG["ignored_versions"]
diff --git a/optimization/snippets/sync_api_test.py b/optimization/snippets/sync_api_test.py
index a5354e18c3be..6bfffc0c7c86 100644
--- a/optimization/snippets/sync_api_test.py
+++ b/optimization/snippets/sync_api_test.py
@@ -15,7 +15,8 @@
 import google.auth
 import pytest
 
-from samples.snippets import sync_api
+
+import sync_api
 
 
 def test_call_sync_api(capsys: pytest.LogCaptureFixture) -> None:
diff --git a/optimization/snippets/sync_api_with_long_timeout_test.py b/optimization/snippets/sync_api_with_long_timeout_test.py
index 197edd34ba7e..4955fa47cb4e 100644
--- a/optimization/snippets/sync_api_with_long_timeout_test.py
+++ b/optimization/snippets/sync_api_with_long_timeout_test.py
@@ -15,7 +15,7 @@
 import google.auth
 import pytest
 
-from samples.snippets import sync_api_with_long_timeout
+import sync_api_with_long_timeout
 
 
 def test_long_timeout(capsys: pytest.LogCaptureFixture) -> None: