diff --git a/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1/test_datastream.py b/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1/test_datastream.py
index 2c4e298d5d53..95019d25e97d 100644
--- a/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1/test_datastream.py
+++ b/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1/test_datastream.py
@@ -50,6 +50,7 @@
 from google.protobuf import timestamp_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
 
diff --git a/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1alpha1/test_datastream.py b/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1alpha1/test_datastream.py
index 6dcd42284355..e0d92949d8f4 100644
--- a/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1alpha1/test_datastream.py
+++ b/packages/google-cloud-datastream/tests/unit/gapic/datastream_v1alpha1/test_datastream.py
@@ -46,6 +46,7 @@
 from google.protobuf import timestamp_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