fix gcode ref

KalicoCrew · Dec 21, 2023 · b147871 · b147871
1 parent 4d30922
commit b147871
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/klippy/klippy.py b/klippy/klippy.py
@@ -379,13 +379,11 @@ def wait_while(self, condition_cb):
         waits until callback returns False
             (or is interrupted, or printer shuts down)
         """
-        counter = self.gcode.get_interrupt_counter()
+        gcode = self.lookup_object("gcode")
+        counter = gcode.get_interrupt_counter()
         eventtime = self.reactor.monotonic()
         while condition_cb(eventtime):
-            if (
-                self.is_shutdown()
-                or counter != self.gcode.get_interrupt_counter()
-            ):
+            if self.is_shutdown() or counter != gcode.get_interrupt_counter():
                 return
             eventtime = self.reactor.pause(eventtime + 1.0)