diff --git a/tests/unit/gapic/functions_v1/test_cloud_functions_service.py b/tests/unit/gapic/functions_v1/test_cloud_functions_service.py
index 1f6f24f..0982e37 100644
--- a/tests/unit/gapic/functions_v1/test_cloud_functions_service.py
+++ b/tests/unit/gapic/functions_v1/test_cloud_functions_service.py
@@ -50,6 +50,7 @@
 from google.type import expr_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/functions_v2/test_function_service.py b/tests/unit/gapic/functions_v2/test_function_service.py
index f1b2361..a678bfa 100644
--- a/tests/unit/gapic/functions_v2/test_function_service.py
+++ b/tests/unit/gapic/functions_v2/test_function_service.py
@@ -49,6 +49,7 @@
 from google.protobuf import timestamp_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