diff --git a/tests/unit/gapic/data_v1alpha/test_alpha_analytics_data.py b/tests/unit/gapic/data_v1alpha/test_alpha_analytics_data.py
index fc39619..561a750 100644
--- a/tests/unit/gapic/data_v1alpha/test_alpha_analytics_data.py
+++ b/tests/unit/gapic/data_v1alpha/test_alpha_analytics_data.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.analytics.data_v1alpha.services.alpha_analytics_data import (
     AlphaAnalyticsDataAsyncClient,
diff --git a/tests/unit/gapic/data_v1beta/test_beta_analytics_data.py b/tests/unit/gapic/data_v1beta/test_beta_analytics_data.py
index 00f70a6..1b55e9b 100644
--- a/tests/unit/gapic/data_v1beta/test_beta_analytics_data.py
+++ b/tests/unit/gapic/data_v1beta/test_beta_analytics_data.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.analytics.data_v1beta.services.beta_analytics_data import (
     BetaAnalyticsDataAsyncClient,