diff --git a/powerwall.sh b/powerwall.sh
index a6606673..008651af 100755
--- a/powerwall.sh
+++ b/powerwall.sh
@@ -1858,7 +1858,7 @@ HOSTMGMT(){
 			ESXI_HOST=${PW_HOST_ADDRESSES[iDS-vMS-Host${HOST}]}
 		elif [ "${PW_HOST_ADDRESSES[${HOST}]}" != "" ]; then
 			ESXI_HOST=${PW_HOST_ADDRESSES[${HOST}]}
-		elif [ "${1^^}" == "ALL" ] || [ "${1^^}" == "MAIN" ] || [ "${1^^}" == "SYS" ] || [ "${1^^}" == "OFFSITE" ]; then
+		elif [ "${1^^}" == "ALL" ] || [ "${1^^}" == "MAINSITE" ] || [ "${1^^}" == "OFFSITE" ] || [ "${1^^}" == "MAIN" ] || [ "${1^^}" == "SYS" ]; then
 			ESXI_HOST=${1^^}
 		else
 			ESXI_HOST=''
@@ -1900,7 +1900,7 @@ HOSTMGMT(){
 	([ "${HOST}" == "" ] || ([ "${POWER}" == "" ] && [ "${ESXI}" == "" ])) && echo -e "Invalid command arguments\n" && exit 1
 	
 	if [ "${MODE}" == "esxi" ]; then
-		if [ "${ESXI_HOST}" == "ALL" ] || [ "${ESXI_HOST}" == "MAIN" ] || [ "${ESXI_HOST}" == "SYS" ] || [ "${ESXI_HOST}" == "OFFSITE" ]; then
+		if [ "${1^^}" == "ALL" ] || [ "${1^^}" == "MAINSITE" ] || [ "${1^^}" == "OFFSITE" ] || [ "${1^^}" == "MAIN" ] || [ "${1^^}" == "SYS" ]; then
 			if [ "${ESXI}" == "maint" ] || [ "${ESXI}" == "maintenance" ] || [ "${ESXI}" == "mm" ]; then
 				echo -e "${idsCL[LightGreen]}PUTTING '${ESXI_HOST}' HOSTS INTO MAINTENANCE MODE${idsCL[Default]}\n"
 				SHUTDOWN_SERVERS ${ESXI_HOST,,} maintenance