diff --git a/yarn-project/publish_npm.sh b/yarn-project/publish_npm.sh
index acc590c861cd..8d657c098b2f 100755
--- a/yarn-project/publish_npm.sh
+++ b/yarn-project/publish_npm.sh
@@ -58,13 +58,11 @@ function deploy_package() {
       jq --arg v $VERSION ".dependencies[\"$PKG\"] = \$v" package.json >$TMP && mv $TMP package.json
     done
 
-    # TODO: Remove this after @noir package resolution is fixed
-    if [[ "$PACKAGE_NAME" == "@aztec/pxe" ]]; then
-      # Hardcodes "1.0.0-beta.1" for @noir-lang/types
-      for PKG in $(jq --raw-output ".dependencies | keys[] | select(. == \"@noir-lang/types\")" package.json); do
-        jq ".dependencies[\"$PKG\"] = \"1.0.0-beta.1\"" package.json >$TMP && mv $TMP package.json
-      done
-    fi
+    # TODO: Remove this after @noir-lang package resolution is fixed
+    # Hardcodes "1.0.0-beta.1" for @noir-lang packages
+    for PKG in $(jq --raw-output ".dependencies | keys[] | select(contains(\"@noir-lang/\"))" package.json); do
+      jq ".dependencies[\"$PKG\"] = \"1.0.0-beta.1\"" package.json >$TMP && mv $TMP package.json
+    done
   fi
 
   # Publish