From b85168845564f219ab833f748d5d23355d0ba5e9 Mon Sep 17 00:00:00 2001
From: Bea Nance <bwnance@gmail.com>
Date: Wed, 17 Jul 2024 11:40:03 -0400
Subject: [PATCH] lift after done

---
 klippy/extras/probe.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/klippy/extras/probe.py b/klippy/extras/probe.py
index f26cdd2b1..de006d699 100644
--- a/klippy/extras/probe.py
+++ b/klippy/extras/probe.py
@@ -538,12 +538,13 @@ def _lift_toolhead(self):
     def _move_next(self):
         toolhead = self.printer.lookup_object("toolhead")
         # Check if done probing
+        done = False
         if len(self.results) >= len(self.probe_points):
             toolhead.get_last_move_time()
             res = self.finalize_callback(self.probe_offsets, self.results)
             if isinstance(res, (int, float)):
                 if res == 0:
-                    return True
+                    done = True
                 if self.adaptive_horizontal_move_z:
                     # then res is error
                     error = math.ceil(res)
@@ -552,9 +553,11 @@ def _move_next(self):
                         self.min_horizontal_move_z,
                     )
             elif res != "retry":
-                return True
+                done = True
             self.results = []
         self._lift_toolhead()
+        if done:
+            return True
         # Move to next XY probe point
         nextpos = list(self.probe_points[len(self.results)])
         if self.use_offsets: