diff --git a/tfjs-core/scripts/publish-npm.sh b/tfjs-core/scripts/publish-npm.sh
index 6dcf60159a2..41ebdbceae4 100755
--- a/tfjs-core/scripts/publish-npm.sh
+++ b/tfjs-core/scripts/publish-npm.sh
@@ -35,8 +35,8 @@ if [ "$BRANCH" != "master" ]; then
   exit
 fi
 
-if ! [[ "$ORIGIN" =~ tensorflow/tfjs-core ]]; then
-  echo "Error: Switch to the main repo (tensorflow/tfjs-core) before publishing."
+if ! [[ "$ORIGIN" =~ tensorflow/tfjs ]]; then
+  echo "Error: Switch to the main repo (tensorflow/tfjs) before publishing."
   exit
 fi