diff --git a/pkgs/development/python-modules/expecttest/default.nix b/pkgs/development/python-modules/expecttest/default.nix
index e9cdcf3f5745..0d52c20ba8f0 100644
--- a/pkgs/development/python-modules/expecttest/default.nix
+++ b/pkgs/development/python-modules/expecttest/default.nix
@@ -7,19 +7,19 @@
   pytestCheckHook,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage {
   pname = "expecttest";
-  version = "0.1.4";
-  format = "pyproject";
+  version = "0.2.1";
+  pyproject = true;
 
   src = fetchFromGitHub {
-    owner = "ezyang";
-    repo = pname;
-    rev = "v${version}";
-    hash = "sha256-BFM0jFWXfH72n9XuFtU9URW8LWGPVJncXniBV5547W4=";
+    owner = "pytorch";
+    repo = "expecttest";
+    rev = "683b09a352cc426851adc2e3a9f46e0ab25e4dee"; # no tags
+    hash = "sha256-e9/KxPN/w0mrFYgesRGqaiDQ6gor7BpX/5/B0NPtXLY=";
   };
 
-  nativeBuildInputs = [ poetry-core ];
+  build-system = [ poetry-core ];
 
   nativeCheckInputs = [
     hypothesis
@@ -32,7 +32,7 @@ buildPythonPackage rec {
     maintainers = [ lib.maintainers.SomeoneSerge ];
     license = lib.licenses.mit;
     description = ''EZ Yang "golden" tests (testing against a reference implementation)'';
-    homepage = "https://github.com/ezyang/expecttest";
+    homepage = "https://github.com/pytorch/expecttest";
     platforms = lib.platforms.unix;
   };
 }