diff --git a/tests/unit/gapic/dataform_v1beta1/test_dataform.py b/tests/unit/gapic/dataform_v1beta1/test_dataform.py
index 6e29c64..742b833 100644
--- a/tests/unit/gapic/dataform_v1beta1/test_dataform.py
+++ b/tests/unit/gapic/dataform_v1beta1/test_dataform.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