diff --git a/pkgs/development/python-modules/vqgan-jax/default.nix b/pkgs/development/python-modules/vqgan-jax/default.nix
index ba410f6f1ea6..5c76aa7d2c68 100644
--- a/pkgs/development/python-modules/vqgan-jax/default.nix
+++ b/pkgs/development/python-modules/vqgan-jax/default.nix
@@ -12,6 +12,9 @@
 buildPythonPackage {
   pname = "vqgan-jax";
   version = "unstable-2022-04-20";
+  pyproject = true;
+
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "patil-suraj";
@@ -22,8 +25,6 @@ buildPythonPackage {
 
   build-system = [ setuptools ];
 
-  pyproject = true;
-
   buildInputs = [ jaxlib ];
 
   dependencies = [