diff --git a/powerwall.sh b/powerwall.sh
index 51e72699..85c16eb6 100755
--- a/powerwall.sh
+++ b/powerwall.sh
@@ -536,9 +536,7 @@ CHECKPOWER(){
 	while [ $# -gt 0 ]; do
 		case "$1" in
 			-avg) avgdays=${2};;
-			hosts) PTYPE=hosts;;
-			mainsite) PTYPE=mainsite;;
-			offsite) PTYPE=offsite;;
+			-type) PTYPE=${2};;
 		esac
 		shift
 	done