diff --git a/karton/core/karton.py b/karton/core/karton.py
index 3001428..f7be257 100644
--- a/karton/core/karton.py
+++ b/karton/core/karton.py
@@ -437,7 +437,7 @@ def _send_signaling_status_task_begin(self, task: Task) -> None:
         self._send_signaling_status_task("task_begin")
 
     def _send_signaling_status_task_end(
-        self, task: Task, ex: Optional[Exception]
+        self, task: Task, ex: Optional[KartonTaskException]
     ) -> None:
         """Send a begin status signaling task.