diff --git a/pkgs/development/python-modules/tensorflow/2/default.nix b/pkgs/development/python-modules/tensorflow/2/default.nix
index bcf693691785..68c983c05941 100644
--- a/pkgs/development/python-modules/tensorflow/2/default.nix
+++ b/pkgs/development/python-modules/tensorflow/2/default.nix
@@ -110,7 +110,7 @@ let
       # Fixes for NixOS jsoncpp
       ../system-jsoncpp.patch
 
-      ./lift-gast-restriction.patch
+      ./relax-dependencies.patch
 
       # see https://github.com/tensorflow/tensorflow/issues/40688
       (fetchpatch {
diff --git a/pkgs/development/python-modules/tensorflow/2/lift-gast-restriction.patch b/pkgs/development/python-modules/tensorflow/2/relax-dependencies.patch
similarity index 53%
rename from pkgs/development/python-modules/tensorflow/2/lift-gast-restriction.patch
rename to pkgs/development/python-modules/tensorflow/2/relax-dependencies.patch
index 7b1b3b135bb0..fc11a04cbdc5 100644
--- a/pkgs/development/python-modules/tensorflow/2/lift-gast-restriction.patch
+++ b/pkgs/development/python-modules/tensorflow/2/relax-dependencies.patch
@@ -1,11 +1,16 @@
 diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
-index 992f2eae22..d9386f9b13 100644
+index 594e74f40c0..bfbf010144f 100644
 --- a/tensorflow/tools/pip_package/setup.py
 +++ b/tensorflow/tools/pip_package/setup.py
-@@ -56,5 +56,5 @@ REQUIRED_PACKAGES = [
+@@ -54,9 +54,9 @@ _VERSION = '2.3.1'
+ REQUIRED_PACKAGES = [
+     'absl-py >= 0.7.0',
      'astunparse == 1.6.3',
 -    'gast == 0.3.3',
 +    'gast >= 0.3.3',
      'google_pasta >= 0.1.8',
-     'h5py >= 2.10.0, < 2.11.0',
+-    'h5py >= 2.10.0, < 2.11.0',
++    'h5py >= 2.10.0',
      'keras_preprocessing >= 1.1.1, < 1.2',
+     # TODO(mihaimaruseac): numpy 1.19.0 has ABI breakage
+     # https://github.com/numpy/numpy/pull/15355