diff --git a/pkgs/development/python-modules/flax/default.nix b/pkgs/development/python-modules/flax/default.nix
index 119b841cde81..89f24260f486 100644
--- a/pkgs/development/python-modules/flax/default.nix
+++ b/pkgs/development/python-modules/flax/default.nix
@@ -13,7 +13,6 @@
   numpy,
   optax,
   orbax-checkpoint,
-  orbax-export,
   pyyaml,
   rich,
   tensorstore,
@@ -36,14 +35,15 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "flax";
-  version = "0.12.6";
+  version = "0.12.7";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "google";
     repo = "flax";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-rIDfF9W8cxF0njH4e4uhqURQ0C4N8Boe76u6meMgC34=";
+    hash = "sha256-a78KiTsCCARWZvbxz9QKdUKnjkDJGXcPVVJu5rU4m/U=";
   };
 
   build-system = [
@@ -58,7 +58,6 @@ buildPythonPackage (finalAttrs: {
     numpy
     optax
     orbax-checkpoint
-    orbax-export
     pyyaml
     rich
     tensorstore
@@ -78,11 +77,6 @@ buildPythonPackage (finalAttrs: {
     tensorflow
   ];
 
-  pytestFlags = [
-    # FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar.
-    "-Wignore::FutureWarning"
-  ];
-
   disabledTestPaths = [
     # Docs test, needs extra deps + we're not interested in it.
     "docs/_ext/codediff_test.py"