diff --git a/packages/google-cloud-trace/tests/unit/gapic/trace_v1/test_trace_service.py b/packages/google-cloud-trace/tests/unit/gapic/trace_v1/test_trace_service.py
index 5f3dc0ac3c65..9bda9a496319 100644
--- a/packages/google-cloud-trace/tests/unit/gapic/trace_v1/test_trace_service.py
+++ b/packages/google-cloud-trace/tests/unit/gapic/trace_v1/test_trace_service.py
@@ -27,7 +27,7 @@
 import math
 import pytest
 from proto.marshal.rules.dates import DurationRule, TimestampRule
-
+from proto.marshal.rules import wrappers
 
 from google.api_core import client_options
 from google.api_core import exceptions as core_exceptions
diff --git a/packages/google-cloud-trace/tests/unit/gapic/trace_v2/test_trace_service.py b/packages/google-cloud-trace/tests/unit/gapic/trace_v2/test_trace_service.py
index 4228a07e1590..9c85479b2d07 100644
--- a/packages/google-cloud-trace/tests/unit/gapic/trace_v2/test_trace_service.py
+++ b/packages/google-cloud-trace/tests/unit/gapic/trace_v2/test_trace_service.py
@@ -27,7 +27,7 @@
 import math
 import pytest
 from proto.marshal.rules.dates import DurationRule, TimestampRule
-
+from proto.marshal.rules import wrappers
 
 from google.api_core import client_options
 from google.api_core import exceptions as core_exceptions