diff --git a/SlicerExtension/LiveUltrasoundAi/LumpNavAISimulation/LumpNavAISimulation.py b/SlicerExtension/LiveUltrasoundAi/LumpNavAISimulation/LumpNavAISimulation.py
index 1d33b62..d6aad60 100644
--- a/SlicerExtension/LiveUltrasoundAi/LumpNavAISimulation/LumpNavAISimulation.py
+++ b/SlicerExtension/LiveUltrasoundAi/LumpNavAISimulation/LumpNavAISimulation.py
@@ -761,7 +761,7 @@ def runTrajectoryAnalysis(self, start, stop) -> int:
             self.processing = False
             return exitCode
         
-    def plotCauteryTrajectory(self, start, stop, clean) -> None:
+    def plotCauteryTrajectory(self, start, stop, clean) -> int:
         logging.info("Plotting cautery trajectory")
         self.processing = True