diff --git a/inc/sites.inc b/inc/sites.inc
index ce92d450..4922afe0 100755
--- a/inc/sites.inc
+++ b/inc/sites.inc
@@ -582,8 +582,8 @@ SITEINFO(){
 				echo -en "\n\033[K${idsCL[LightCyan]}Enter the line number to edit, or (s) to save: ${idsCL[Default]}"
 				read edit
 				
-			    case "$1" in
-			        4) [ "${hsts}" == "-" ] && hsts="Yes" || hsts='-';;
+			    case "${edit}" in
+			        4) [ "${hsts}" == "-" ] && hsts='Yes' || hsts='-';;
 			    esac
 				
 				echo -e "\033[11A"