diff --git a/tests/unit/test_rest_streaming.py b/tests/unit/test_rest_streaming.py
index 1b34a04f..a44c83c0 100644
--- a/tests/unit/test_rest_streaming.py
+++ b/tests/unit/test_rest_streaming.py
@@ -28,6 +28,7 @@
 from google.protobuf import timestamp_pb2
 
 
+__protobuf__ = proto.module(package=__name__)
 SEED = int(time.time())
 logging.info(f"Starting rest streaming tests with random seed: {SEED}")
 random.seed(SEED)