diff --git a/morpheus/pipeline/__init__.py b/morpheus/pipeline/__init__.py
index 01257291e0..6984c702b5 100644
--- a/morpheus/pipeline/__init__.py
+++ b/morpheus/pipeline/__init__.py
@@ -22,7 +22,7 @@
 from morpheus.pipeline.stage_schema import StageSchema
 from morpheus.pipeline.sender import Sender
 from morpheus.pipeline.receiver import Receiver
-from morpheus.pipeline.stream_wrapper import BaseStage
+from morpheus.pipeline.base_stage import BaseStage
 from morpheus.pipeline.stage import Stage
 from morpheus.pipeline.single_port_stage import SinglePortStage
 from morpheus.pipeline.multi_message_stage import MultiMessageStage
diff --git a/morpheus/pipeline/stream_wrapper.py b/morpheus/pipeline/base_stage.py
similarity index 100%
rename from morpheus/pipeline/stream_wrapper.py
rename to morpheus/pipeline/base_stage.py
diff --git a/morpheus/pipeline/pipeline.py b/morpheus/pipeline/pipeline.py
index ffcfc1b526..5acd1a0245 100644
--- a/morpheus/pipeline/pipeline.py
+++ b/morpheus/pipeline/pipeline.py
@@ -34,7 +34,7 @@
 from morpheus.pipeline.sender import Sender
 from morpheus.pipeline.source_stage import SourceStage
 from morpheus.pipeline.stage import Stage
-from morpheus.pipeline.stream_wrapper import BaseStage
+from morpheus.pipeline.base_stage import BaseStage
 from morpheus.utils.type_utils import pretty_print_type_name
 
 logger = logging.getLogger(__name__)
diff --git a/morpheus/pipeline/stage_schema.py b/morpheus/pipeline/stage_schema.py
index c3653e15fa..696510c683 100644
--- a/morpheus/pipeline/stage_schema.py
+++ b/morpheus/pipeline/stage_schema.py
@@ -16,7 +16,7 @@
 import typing
 
 if typing.TYPE_CHECKING:
-    from .stream_wrapper import BaseStage
+    from .base_stage import BaseStage
 
 
 class PortSchema: