diff --git a/scripts/publish-npm.sh b/scripts/publish-npm.sh
index c819ab3845..59dc0e141a 100755
--- a/scripts/publish-npm.sh
+++ b/scripts/publish-npm.sh
@@ -34,7 +34,7 @@ if [ "$BRANCH" != "master" ]; then
   exit
 fi
 
-if ! [[ "$ORIGIN" =~ ^https://github.com/tensorflow/tfjs-core ]]; then
+if ! [[ "$ORIGIN" =~ tensorflow/tfjs-core ]]; then
   echo "Error: Switch to the main repo (tensorflow/tfjs-core) before publishing."
   exit
 fi