diff --git a/plex-token.sh b/plex-token.sh
index 3ad831a..456acc7 100644
--- a/plex-token.sh
+++ b/plex-token.sh
@@ -60,7 +60,8 @@ token_from_api() {
   echo $X_PLEX_TOKEN
 }
 
-token_from_Pref_XML
-token_from_api
-
-exit
+case "$1" in
+    api) token_from_api;;
+    ""|xml) token_from_Pref_XML;;
+    *) echo "'$1' is not a valid plex-tools token command";;
+esac