diff --git a/python/PiFinder/ui/preview.py b/python/PiFinder/ui/preview.py
index 35391ddc..75655ba9 100644
--- a/python/PiFinder/ui/preview.py
+++ b/python/PiFinder/ui/preview.py
@@ -247,7 +247,6 @@ def key_number(self, number):
                 self.shared_state.set_solve_pixel((256, 256))
                 self.config_object.set_option("solve_pixel", (256, 256))
                 self.align_mode = False
-                self.update(force=True)
             if number in list(range(1, self.highlight_count + 1)):
                 # They picked a star to align....
                 star_index = number - 1
@@ -260,4 +259,6 @@ def key_number(self, number):
                         (star_cam_x, star_cam_y),
                     )
                 self.align_mode = False
-                self.update(force=True)
+
+            self.shared_state.set_camera_align(self.align_mode)
+            self.update(force=True)