diff --git a/pkgs/development/python-modules/keras/default.nix b/pkgs/development/python-modules/keras/default.nix
index ac5d47623a8ac5..4c1c520e436643 100644
--- a/pkgs/development/python-modules/keras/default.nix
+++ b/pkgs/development/python-modules/keras/default.nix
@@ -45,7 +45,7 @@ buildPythonPackage rec {
     dm-tree
     h5py
     markdown-it-py
-    ml-dtypes
+    ({ ml-dtypesTF = ml-dtypes; } // tensorflow).ml-dtypesTF
     namex
     numpy
     optree