diff --git a/streaming/__init__.py b/streaming/__init__.py
index e7b98dcca..b8f3e2541 100644
--- a/streaming/__init__.py
+++ b/streaming/__init__.py
@@ -6,6 +6,7 @@
 import streaming.multimodal as multimodal
 import streaming.text as text
 import streaming.vision as vision
+from streaming._version import __version__  # noqa: F401
 from streaming.base import (CSVWriter, JSONWriter, LocalDataset, MDSWriter, Stream,
                             StreamingDataLoader, StreamingDataset, TSVWriter, XSVWriter)
 
diff --git a/tests/conftest.py b/tests/conftest.py
index adf131628..191b50af9 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -9,6 +9,10 @@
 import pytest
 from moto import mock_aws
 
+from tests.common.utils import compressed_local_remote_dir  # pyright: ignore # noqa: F401
+from tests.common.utils import get_free_tcp_port  # pyright: ignore # noqa: F401
+from tests.common.utils import local_remote_dir  # pyright: ignore # noqa: F401
+
 MY_BUCKET = 'streaming-test-bucket'
 MY_PREFIX = 'train'
 GCS_URL = 'https://storage.googleapis.com'