diff --git a/setup.py b/setup.py
index 426b60c..9f647a2 100644
--- a/setup.py
+++ b/setup.py
@@ -37,9 +37,11 @@
 # Unfortunately, there's no tensorflowjs package on pypi for aarch64,
 # as far as I can tell.  Also, it seems that the aarch64 build is
 # quite picky about its numpy version, and you have to let it pull in
-# the exact version that tensorflow was built with.
+# the exact version that tensorflow was built with. We also avoid setting
+# the tensorflowjs version to allow looking for the correct match.
 if platform.machine() not in ["aarch64", "arm64"]:
-    deps += ["tensorflowjs>=4.16,<4.17", "numpy>=1.26.3,<1.27"]
+    deps += ["tensorflowjs",
+             "numpy>=1.26.3,<1.27"]
 
 # Get the long description from the relevant file
 with open(os.path.join(here, "README.md"), "r") as f: