diff --git a/index.php b/index.php
index 20d8335..05d2374 100644
--- a/index.php
+++ b/index.php
@@ -138,7 +138,7 @@
     $sites = $unifidata->list_sites();
     $_SESSION['sites'] = $sites;
 } else {
-$sites = $_SESSION['sites'];
+    $sites = $_SESSION['sites'];
 }
 
 /*
@@ -148,6 +148,8 @@
     $site_info = $unifidata->stat_sysinfo();
     $detected_controller_version = $site_info[0]->version;
     $_SESSION['detected_controller_version'] = $detected_controller_version;
+} else {
+    $detected_controller_version = $_SESSION['detected_controller_version'];
 }
 
 /*