diff --git a/installer/install.sh b/installer/install.sh
index cf194b8..047581a 100755
--- a/installer/install.sh
+++ b/installer/install.sh
@@ -600,13 +600,18 @@ build_binaries() {
     echo "Building from: $(pwd)"
     
     # Create cmd directory structure if it doesn't exist
+    mkdir -p ./cmd/atlas-api
+    mkdir -p ./cmd/atlas-tui
+    
+    # Always update main.go files to ensure they're current
     if [[ ! -f "./cmd/atlas-api/main.go" ]]; then
         echo -e "${YELLOW}cmd/ directory not found, creating...${NC}"
-        mkdir -p ./cmd/atlas-api
-        mkdir -p ./cmd/atlas-tui
-        
-        # Create atlas-api/main.go
-        cat > ./cmd/atlas-api/main.go <<'MAINGO'
+    else
+        echo "  Updating cmd/atlas-api/main.go..."
+    fi
+    
+    # Create/update atlas-api/main.go
+    cat > ./cmd/atlas-api/main.go <<'MAINGO'
 package main
 
 import (
@@ -749,6 +754,8 @@ func main() {
 MAINGO
         
         echo -e "${GREEN}Created cmd/ directory structure${NC}"
+    else
+        echo "  Updating cmd/atlas-tui/main.go..."
     fi
     
     # Verify cmd files exist now