diff --git a/src/interface/desktop/configure_screen.py b/src/interface/desktop/configure_screen.py
index e21b761a..2000dff5 100644
--- a/src/interface/desktop/configure_screen.py
+++ b/src/interface/desktop/configure_screen.py
@@ -15,8 +15,9 @@ class ConfigureScreen(QtWidgets.QDialog):
     3. Save the configuration to khoj.yml and start the server
     """
 
-    def __init__(self, parent=None):
+    def __init__(self, config_file, parent=None):
         super(ConfigureScreen, self).__init__(parent=parent)
+        self.config_file = config_file
 
         # Initialize Configure Window
         self.setWindowFlags(Qt.WindowType.WindowStaysOnTopHint)
diff --git a/src/main.py b/src/main.py
index b35ddbbe..99fd3036 100644
--- a/src/main.py
+++ b/src/main.py
@@ -24,16 +24,16 @@ app.include_router(router)
 
 
 def run():
+    # Load config from CLI
+    args = cli(sys.argv[1:])
+
     # Setup Base GUI
     gui = QtWidgets.QApplication([])
     gui.setQuitOnLastWindowClosed(False)
-    configure_screen = ConfigureScreen()
+    configure_screen = ConfigureScreen(args.config_file)
     tray = create_system_tray(gui, configure_screen)
     tray.show()
 
-    # Load config from CLI
-    args = cli(sys.argv[1:])
-
     # Trigger First Run Experience, if required
     if args.config is None:
         configure_screen.show()