diff --git a/core/amber/src/main/python/core/architecture/packaging/input_manager.py b/core/amber/src/main/python/core/architecture/packaging/input_manager.py
index 519ab448628..4a50fccea81 100644
--- a/core/amber/src/main/python/core/architecture/packaging/input_manager.py
+++ b/core/amber/src/main/python/core/architecture/packaging/input_manager.py
@@ -89,7 +89,7 @@ def process_data_payload(
     ) -> Iterator[Union[Tuple, InternalMarker]]:
         # special case used to yield for source op
         if from_ == InputManager.SOURCE_STARTER:
-            yield EndOfInputChannel()
+            yield EndOfInputPort()
             yield EndOfOutputPorts()
             return