diff --git a/src/plugins/timediff/time_diff_menu_action.py b/src/plugins/timediff/time_diff_menu_action.py
index 3af0fe4..eb94145 100644
--- a/src/plugins/timediff/time_diff_menu_action.py
+++ b/src/plugins/timediff/time_diff_menu_action.py
@@ -1,12 +1,8 @@
-from PySide6.QtWidgets import QWidget, QPushButton, QLabel, QComboBox, QLineEdit, QSizePolicy
+from PySide6.QtWidgets import QWidget, QPushButton
 from typing import Callable
 from src.plugins.domain.raction import RAction
 from src.plugins.timediff.time_diff_menu_widget import TimeDiffMenuWidget
 from src.plugins.timediff.time_diff_pre_process_lines_hook import TimeDiffPreProcessLinesHook
-from src.ui.hbox import HBox
-from src.ui.icon import Icon
-
-from src.i18n import _
 
 
 class TimeDiffMenuAction(RAction):
@@ -23,5 +19,6 @@ class TimeDiffMenuAction(RAction):
     def _triggered(self, active: bool, delta: float):
         self.time_diff_hook.active = active
         self.time_diff_hook.delta = delta
-        print(f"time_diff_hook updated: {active} -- {delta}")
-        self.after_action()
+        # print(f"time_diff_hook updated: {active} -- {delta}")
+        if self.after_action:
+            self.after_action()