diff --git a/tests/unit/gapic/dataqna_v1alpha/test_auto_suggestion_service.py b/tests/unit/gapic/dataqna_v1alpha/test_auto_suggestion_service.py
index a05c152..3768889 100644
--- a/tests/unit/gapic/dataqna_v1alpha/test_auto_suggestion_service.py
+++ b/tests/unit/gapic/dataqna_v1alpha/test_auto_suggestion_service.py
@@ -33,6 +33,7 @@
 from google.oauth2 import service_account
 import grpc
 from grpc.experimental import aio
+from proto.marshal.rules import wrappers
 from proto.marshal.rules.dates import DurationRule, TimestampRule
 import pytest
 
diff --git a/tests/unit/gapic/dataqna_v1alpha/test_question_service.py b/tests/unit/gapic/dataqna_v1alpha/test_question_service.py
index c42222a..5755e22 100644
--- a/tests/unit/gapic/dataqna_v1alpha/test_question_service.py
+++ b/tests/unit/gapic/dataqna_v1alpha/test_question_service.py
@@ -37,6 +37,7 @@
 from google.rpc import status_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