diff --git a/src/tools/visualization.py b/src/tools/visualization.py
index 30d2554..46079a1 100644
--- a/src/tools/visualization.py
+++ b/src/tools/visualization.py
@@ -57,6 +57,7 @@ class Window:
         self.__canvas.scale(tk.ALL, 0, 0, scale, scale)
         self.__canvas.update()
 
-    def done(self) -> None:
-        self.realign()
+    def done(self, realign: bool = True) -> None:
+        if realign:
+            self.realign()
         self.__canvas.mainloop()