diff --git a/guiheader.py b/guiheader.py
index 5158703..472cac0 100644
--- a/guiheader.py
+++ b/guiheader.py
@@ -21,7 +21,7 @@ class GUIHeader(QWidget):
         self.title = parent.title
         self.settings = parent.settings
         self.user_mode = user_mode
-        self.current_user_mode = user_mode
+        self.current_user_mode_name = user_mode.name
         self.extended = extended
         self.cafe = parent.cafe
         self.cyca = parent.cyca
@@ -191,12 +191,13 @@ class GUIHeader(QWidget):
         title_name =  user_mode.name
         dry_run_tags = ['Emittance', 'Dry']
         if any([x in self.header_wgt.title() for x in dry_run_tags]):
-            title_name = 'DRY RUN' if user_mode.name == 'SIMULATION' else user_mode.name
+            title_name = 'DRY RUN' if user_mode.name == 'SIMULATION' \
+                         else user_mode.name
         self.header_wgt.setObjectName(user_mode.name)
         self.header_wgt.setTitle(self.header_wgt.title().replace(
-            self.current_user_mode.name, title_name))
+            self.current_user_mode_name, title_name))
         self.header_wgt.style().polish(self.header_wgt)
-        self.current_user_mode = user_mode
+        self.current_user_mode_name = title_name
         
 
     def beamline_widget(self, beamline="Aramis"):