diff --git a/ml/synthesis/pyproject.toml b/ml/synthesis/pyproject.toml
index 7ada3e3cf..f47809f30 100644
--- a/ml/synthesis/pyproject.toml
+++ b/ml/synthesis/pyproject.toml
@@ -20,7 +20,7 @@ pandas = { version = "2.0.2", extras = ["performance"] }
 tabulate = "^0.9.0"
 
 tensorflow = "2.13.1"
-tensorflow-cpu-aws = { version = "^2.13.1", platform = "linux" }
+tensorflow-cpu = { version = "^2.13.1", platform = "linux" }
 tensorflow-intel = { version = "^2.13.1", platform = "win32" }
 tensorflow-macos = { platform = "darwin", version = "^2.13.1" }
 tensorflow-metal = { platform = "darwin", version = "^1.1.0" }