diff --git a/tests/unit/gapic/apigateway_v1/test_api_gateway_service.py b/tests/unit/gapic/apigateway_v1/test_api_gateway_service.py
index a5a20c8..5bcd313 100644
--- a/tests/unit/gapic/apigateway_v1/test_api_gateway_service.py
+++ b/tests/unit/gapic/apigateway_v1/test_api_gateway_service.py
@@ -45,6 +45,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