diff --git a/src/python/impactx/dashboard/Analyze/plot_PhaseSpaceProjections/phaseSpaceSettings.py b/src/python/impactx/dashboard/Analyze/plot_PhaseSpaceProjections/phaseSpaceSettings.py
index 00045caae..e303adb86 100644
--- a/src/python/impactx/dashboard/Analyze/plot_PhaseSpaceProjections/phaseSpaceSettings.py
+++ b/src/python/impactx/dashboard/Analyze/plot_PhaseSpaceProjections/phaseSpaceSettings.py
@@ -32,7 +32,7 @@ def adjusted_settings_plot(self, num_bins=50, root_rank=0):
     # Matplotlib canvas: figure and plottable axes areas
     fig, axes = plt.subplots(1, 3, figsize=(12, 3))
     (ax_xpx, ax_ypy, ax_tpt) = axes
-    
+
     # Plotting data points if df is not None
     if df is not None:
         # update for plot unit system
diff --git a/src/python/impactx/dashboard/Analyze/plotsMain.py b/src/python/impactx/dashboard/Analyze/plotsMain.py
index 960d3a822..55fcf2e38 100644
--- a/src/python/impactx/dashboard/Analyze/plotsMain.py
+++ b/src/python/impactx/dashboard/Analyze/plotsMain.py
@@ -1,6 +1,7 @@
 import os
 
-from trame.widgets import plotly, vuetify, matplotlib
+from trame.app import get_server
+from trame.widgets import matplotlib, plotly, vuetify
 
 from .analyzeFunctions import AnalyzeFunctions
 from .plot_ParameterEvolutionOverS.overS import line_plot_1d
@@ -54,7 +55,7 @@ def load_dataTable_data():
     """
     Loads and processes data from combined beam and reference particle files.
     """
-    
+
     combined_files = AnalyzeFunctions.combine_files(
         REDUCED_BEAM_DATA, REF_PARTICLE_DATA
     )
@@ -124,7 +125,7 @@ def update_plot():
 
 
 @state.change("selected_headers")
-def on_header_selection_change (selected_headers, **kwargs):
+def on_header_selection_change(selected_headers, **kwargs):
     state.filtered_headers = AnalyzeFunctions.filter_headers(
         state.all_headers, selected_headers
     )
diff --git a/src/python/impactx/dashboard/Toolbar/toolbarMain.py b/src/python/impactx/dashboard/Toolbar/toolbarMain.py
index e6fe1c4b3..9eff43284 100644
--- a/src/python/impactx/dashboard/Toolbar/toolbarMain.py
+++ b/src/python/impactx/dashboard/Toolbar/toolbarMain.py
@@ -187,4 +187,4 @@ def analyze_toolbar():
         ToolbarElements.plot_options()
         ToolbarElements.run_simulation_button()
         # ToolbarElements.export_input_data()
-        # ToolbarElements.switch_theme()
\ No newline at end of file
+        # ToolbarElements.switch_theme()
diff --git a/src/python/impactx/dashboard/__main__.py b/src/python/impactx/dashboard/__main__.py
index 417f7883b..4df17608c 100644
--- a/src/python/impactx/dashboard/__main__.py
+++ b/src/python/impactx/dashboard/__main__.py
@@ -1,7 +1,7 @@
 import sys
 from trame.ui.router import RouterViewLayout
 from trame.ui.vuetify import SinglePageWithDrawerLayout
-from trame.widgets import router, vuetify, xterm
+from trame.widgets import router, vuetify
 
 from .Analyze.plotsMain import AnalyzeSimulation
 from .Input.distributionParameters.distributionMain import DistributionParameters
@@ -9,7 +9,6 @@
 from .Input.latticeConfiguration.latticeMain import LatticeConfiguration
 from .Input.trameFunctions import TrameFunctions
 from .Input.Visualiztion.twiss_phase_space_ellipse.x_px import VisualizeTwiss
-from .Optimize.optimizeMain import Optimize
 from .start import main
 from .Toolbar.toolbarMain import Toolbars