diff --git a/explorer.py b/explorer.py
index 00cc37a..5cae0fc 100644
--- a/explorer.py
+++ b/explorer.py
@@ -2211,7 +2211,9 @@ def choose_trajectory_title(axis_def: dict, axis: str, threshold: float = 0.65)
             if st.button("🔧 Database diagnostiek uitvoeren"):
                 with st.spinner("Bezig met diagnostiek..."):
                     # Import and run diagnostics
-                    from scripts.diagnose_trajectories_cli import diagnose_trajectories
+                    from scripts.diagnose_trajectories_cli import (
+                        run as diagnose_trajectories,
+                    )
 
                     results = diagnose_trajectories(db_path)
                     st.json(results)