diff --git a/scripts/publish_prod.sh b/scripts/publish_prod.sh
index 845cddb3..fdc4c9d6 100755
--- a/scripts/publish_prod.sh
+++ b/scripts/publish_prod.sh
@@ -37,7 +37,7 @@ AWS_PROFILE=govcloud-us1-fed-human-engineering aws sts get-caller-identity
 aws-vault exec prod-engineering -- aws sts get-caller-identity
 
 # Ensure pypi registry access
-read -p "Do you have the PyPi login credentials for datadog account (y/n)?" CONT
+read -p "Do you have access to PyPI (y/n)?" CONT
 if [ "$CONT" != "y" ]; then
     echo "Exiting"
     exit 1
diff --git a/scripts/pypi.sh b/scripts/pypi.sh
index 9d8592c0..6abe685f 100755
--- a/scripts/pypi.sh
+++ b/scripts/pypi.sh
@@ -14,6 +14,9 @@ if [ -d "dist" ]; then
     rm -rf dist;
 fi
 
-# Publish to pypi
-poetry publish --build
+echo "Please enter the PyPI token (password) for datadog-lambda-python"
+echo "This can be found in 1password under the shared-serverless vault"
+read -p "Token name: " TOKEN
 
+# Publish to pypi
+poetry publish --build --username __token__ --password $TOKEN