diff --git a/pkgs/development/python-modules/returns/default.nix b/pkgs/development/python-modules/returns/default.nix
index 098ae334ccd0..0b54d62ba569 100644
--- a/pkgs/development/python-modules/returns/default.nix
+++ b/pkgs/development/python-modules/returns/default.nix
@@ -43,10 +43,6 @@ buildPythonPackage rec {
     typing-extensions
   ];
 
-  preCheck = ''
-    rm -rf returns/contrib/mypy
-  '';
-
   nativeCheckInputs = [
     anyio
     curio
@@ -59,6 +55,14 @@ buildPythonPackage rec {
     trio
   ];
 
+  preCheck = ''
+    rm -rf returns/contrib/mypy
+  '';
+
+  pythonImportsCheck = [
+    "returns"
+  ];
+
   pytestFlagsArray = [
     "--ignore=typesafety"
   ];