diff --git a/pylot/core/util/widgets.py b/pylot/core/util/widgets.py
index 67f80d6f..c77c7d18 100644
--- a/pylot/core/util/widgets.py
+++ b/pylot/core/util/widgets.py
@@ -2248,7 +2248,7 @@ class TuneAutopicker(QWidget):
         else:
             self.disable_autopickTabs()
         try:
-            main_fig.tight_layout()
+            self.fig_dict['main_fig'].tight_layout()
         except:
             pass
         self.figure_tabs.setCurrentIndex(0)