diff --git a/python/morpheus/morpheus/_lib/messages/__init__.pyi b/python/morpheus/morpheus/_lib/messages/__init__.pyi
index 74d7a522d4..e61489d1d6 100644
--- a/python/morpheus/morpheus/_lib/messages/__init__.pyi
+++ b/python/morpheus/morpheus/_lib/messages/__init__.pyi
@@ -37,6 +37,8 @@ class ControlMessage():
     def __init__(self, arg0: ControlMessage) -> None: ...
     @typing.overload
     def __init__(self, arg0: dict) -> None: ...
+    @typing.overload
+    def __init__(self, arg0: object, arg1: dict) -> None: ...
     def add_task(self, task_type: str, task: object | None) -> None: ...
     @typing.overload
     def config(self) -> object | None: ...