diff --git a/installer/install.sh b/installer/install.sh
index 047581a..54f6537 100755
--- a/installer/install.sh
+++ b/installer/install.sh
@@ -711,9 +711,15 @@ func env(key, def string) string {
 	return def
 }
 MAINGO
-        
-        # Create atlas-tui/main.go
-        cat > ./cmd/atlas-tui/main.go <<'MAINGO'
+    
+    # Create/update atlas-tui/main.go
+    if [[ ! -f "./cmd/atlas-tui/main.go" ]]; then
+        echo "  Creating cmd/atlas-tui/main.go..."
+    else
+        echo "  Updating cmd/atlas-tui/main.go..."
+    fi
+    
+    cat > ./cmd/atlas-tui/main.go <<'MAINGO'
 package main
 
 import (
@@ -752,11 +758,8 @@ func main() {
 	}
 }
 MAINGO
-        
-        echo -e "${GREEN}Created cmd/ directory structure${NC}"
-    else
-        echo "  Updating cmd/atlas-tui/main.go..."
-    fi
+    
+    echo -e "${GREEN}cmd/ directory structure ready${NC}"
     
     # Verify cmd files exist now
     if [[ ! -f "./cmd/atlas-api/main.go" ]]; then