diff --git a/packages/google-cloud-automl/tests/unit/gapic/automl_v1/test_auto_ml.py b/packages/google-cloud-automl/tests/unit/gapic/automl_v1/test_auto_ml.py
index 6fb61b1429c2..f5b5c37fbf6b 100644
--- a/packages/google-cloud-automl/tests/unit/gapic/automl_v1/test_auto_ml.py
+++ b/packages/google-cloud-automl/tests/unit/gapic/automl_v1/test_auto_ml.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-automl/tests/unit/gapic/automl_v1/test_prediction_service.py b/packages/google-cloud-automl/tests/unit/gapic/automl_v1/test_prediction_service.py
index c8201905293e..8e496724c3e6 100644
--- a/packages/google-cloud-automl/tests/unit/gapic/automl_v1/test_prediction_service.py
+++ b/packages/google-cloud-automl/tests/unit/gapic/automl_v1/test_prediction_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-automl/tests/unit/gapic/automl_v1beta1/test_auto_ml.py b/packages/google-cloud-automl/tests/unit/gapic/automl_v1beta1/test_auto_ml.py
index be13e6d294d4..abd4770fdd29 100644
--- a/packages/google-cloud-automl/tests/unit/gapic/automl_v1beta1/test_auto_ml.py
+++ b/packages/google-cloud-automl/tests/unit/gapic/automl_v1beta1/test_auto_ml.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-automl/tests/unit/gapic/automl_v1beta1/test_prediction_service.py b/packages/google-cloud-automl/tests/unit/gapic/automl_v1beta1/test_prediction_service.py
index fa6ec1f6c657..2d80be632c40 100644
--- a/packages/google-cloud-automl/tests/unit/gapic/automl_v1beta1/test_prediction_service.py
+++ b/packages/google-cloud-automl/tests/unit/gapic/automl_v1beta1/test_prediction_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