diff --git a/pkgs/development/python-modules/vqgan-jax/default.nix b/pkgs/development/python-modules/vqgan-jax/default.nix
index 3d07a2fc43d1..5c76aa7d2c68 100644
--- a/pkgs/development/python-modules/vqgan-jax/default.nix
+++ b/pkgs/development/python-modules/vqgan-jax/default.nix
@@ -2,6 +2,7 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
+  setuptools,
   flax,
   jax,
   jaxlib,
@@ -11,6 +12,9 @@
 buildPythonPackage {
   pname = "vqgan-jax";
   version = "unstable-2022-04-20";
+  pyproject = true;
+
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "patil-suraj";
@@ -19,11 +23,11 @@ buildPythonPackage {
     hash = "sha256-OZihAXpE0UsgauQ38XDmAF+lrIgz05uK0ro8SCdVsPc=";
   };
 
-  format = "setuptools";
+  build-system = [ setuptools ];
 
   buildInputs = [ jaxlib ];
 
-  propagatedBuildInputs = [
+  dependencies = [
     flax
     jax
     transformers