diff --git a/core/scanner.py b/core/scanner.py
index ea5e095..3bd6a9a 100644
--- a/core/scanner.py
+++ b/core/scanner.py
@@ -586,6 +586,7 @@ class Scanner:
             if self.status in [ScanStatus.FINALIZING, ScanStatus.COMPLETED, ScanStatus.STOPPED]:
                 print(f"\n=== PHASE 2: Running correlation analysis ===")
                 self._run_correlation_phase(max_depth, processed_tasks)
+                self._update_session_state()
 
             # Determine the final status *after* finalization.
             if self._is_stop_requested():
diff --git a/static/js/main.js b/static/js/main.js
index 6a12a87..3a8bd60 100644
--- a/static/js/main.js
+++ b/static/js/main.js
@@ -855,7 +855,7 @@ class DNScopeApp {
                 
                 // Do final graph update when scan completes
                 console.log('Scan completed - performing final graph update');
-                setTimeout(() => this.updateGraph(), 100);
+                setTimeout(() => this.updateGraph(), 1000);
                 break;
                 
             case 'failed':