From 9624afac50c426da7108a2d8cfa605de7d587a1c Mon Sep 17 00:00:00 2001
From: Stephan Yelle <syelle@noreply.localhost>
Date: Sun, 25 Jan 2026 00:27:30 -0500
Subject: [PATCH] Update src/ui.ps1

---
 src/ui.ps1 | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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 }