diff --git a/klippy/toolhead.py b/klippy/toolhead.py
index 2954812b..afb1ef0f 100644
--- a/klippy/toolhead.py
+++ b/klippy/toolhead.py
@@ -613,6 +613,9 @@ class ToolHead:
         return self.trapq
     def register_step_generator(self, handler):
         self.step_generators.append(handler)
+    def unregister_step_generator(self, handler):
+        if handler in self.step_generators:
+            self.step_generators.remove(handler)
     def note_step_generation_scan_time(self, delay, old_delay=0.):
         self.flush_step_generation()
         if old_delay: