diff --git a/upload.sh b/upload.sh
index 3bfd531..73b8fbc 100755
--- a/upload.sh
+++ b/upload.sh
@@ -2,4 +2,16 @@
 cd "$(dirname "$0")"
 rm -rf dist
 pipenv run -- python setup.py sdist bdist_wheel
-pipenv run -- python -m twine upload --repository-url https://test.pypi.org/legacy/ dist/*
+case $1 in
+    prod)
+        pipenv run -- python -m twine upload
+        ;;
+    test)
+        pipenv run -- python -m twine upload --repository-url https://test.pypi.org/legacy/ dist/*
+        ;;
+    *)
+        echo "Either prod or test"
+        exit 1
+        ;;
+esac
+