diff --git a/python/update.py b/python/update.py
index 7ef1fd95b..5a5dab935 100644
--- a/python/update.py
+++ b/python/update.py
@@ -17,10 +17,16 @@ cli.add_command(compile.cli, "compile")
 
 print("Updating deps...")
 os.chdir(os.getenv("BUILD_WORKING_DIRECTORY"))
+
+if pkg := os.getenv("PACKAGE"):
+    upgrade_args = ["--upgrade-package", pkg]
+else:
+    upgrade_args = ["--upgrade"]
+
 sys.argv[1:] = [
     "compile",
     "--allow-unsafe",
-    "--upgrade",
+    *upgrade_args,
     "--no-header",
     "--strip-extras",
     "--generate-hashes",