safe_z_home: Simplify check for axes to home

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 3 insertions, 5 deletions

diff --git a/klippy/extras/safe_z_home.py b/klippy/extras/safe_z_home.py
index 5c21bd3c..4e0801c4 100644
--- a/klippy/extras/safe_z_home.py
+++ b/klippy/extras/safe_z_home.py
@@ -52,11 +52,9 @@ class SafeZHoming:
                     toolhead.get_kinematics().note_z_not_homed()
 
         # Determine which axes we need to home
-        if not any([axis in params.keys() for axis in ['X', 'Y', 'Z']]):
-            need_x, need_y, need_z = [True] * 3
-        else:
-            need_x, need_y, need_z = tuple(axis in params
-                                           for axis in ['X', 'Y', 'Z'])
+        need_x, need_y, need_z = [axis in params for axis in "XYZ"]
+        if not need_x and not need_y and not need_z:
+            need_x = need_y = need_z = True
 
         # Home XY axes if necessary
         new_params = {}