diff --git a/tests/unit/gapic/container_v1/test_cluster_manager.py b/tests/unit/gapic/container_v1/test_cluster_manager.py
index f0fcb333..faedea20 100644
--- a/tests/unit/gapic/container_v1/test_cluster_manager.py
+++ b/tests/unit/gapic/container_v1/test_cluster_manager.py
@@ -38,6 +38,7 @@
 from google.rpc import status_pb2  # type: ignore
 import grpc
 from grpc.experimental import aio
+from proto.marshal.rules import wrappers
 from proto.marshal.rules.dates import DurationRule, TimestampRule
 import pytest
 
diff --git a/tests/unit/gapic/container_v1beta1/test_cluster_manager.py b/tests/unit/gapic/container_v1beta1/test_cluster_manager.py
index 00083844..f9d24a4f 100644
--- a/tests/unit/gapic/container_v1beta1/test_cluster_manager.py
+++ b/tests/unit/gapic/container_v1beta1/test_cluster_manager.py
@@ -38,6 +38,7 @@
 from google.rpc import status_pb2  # type: ignore
 import grpc
 from grpc.experimental import aio
+from proto.marshal.rules import wrappers
 from proto.marshal.rules.dates import DurationRule, TimestampRule
 import pytest