diff --git a/src/ui.ps1 b/src/ui.ps1
index 2cbfa07..85e1cf3 100644
--- a/src/ui.ps1
+++ b/src/ui.ps1
@@ -117,7 +117,11 @@ $subHtml
 function Get-ModuleVersionHtml {
     $mod = Get-Module -ListAvailable -Name SVSMSP | Sort-Object Version -Descending | Select-Object -First 1
 
-    $branchDisplay = switch ($Script:SamyBranch.ToLower()) {
+    $branch = $Script:SamyBranch
+    if ([string]::IsNullOrWhiteSpace($branch)) { $branch = $Script:SamyGitBranch }
+    if ([string]::IsNullOrWhiteSpace($branch)) { $branch = 'beta' }
+
+    $branchDisplay = switch ($branch.ToLower()) {
         'main' { 'Main / Stable' }
         'beta' { 'Beta' }
         default { $Script:SamyBranch }