diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt
index bbfe91f824..715bfb414a 100644
--- a/docker/main/requirements-wheels.txt
+++ b/docker/main/requirements-wheels.txt
@@ -13,7 +13,7 @@ pathvalidate == 3.3.*
 markupsafe == 3.0.*
 python-multipart == 0.0.31
 # Classification Model Training
-tensorflow == 2.19.* ; platform_machine == 'aarch64'
+tensorflow == 2.21.* ; platform_machine == 'aarch64'
 tensorflow-cpu == 2.19.* ; platform_machine == 'x86_64'
 # General
 mypy == 1.6.1