diff --git a/pkgs/development/python-modules/schema-salad/default.nix b/pkgs/development/python-modules/schema-salad/default.nix
index 995c7e206faf..1980b18dab84 100644
--- a/pkgs/development/python-modules/schema-salad/default.nix
+++ b/pkgs/development/python-modules/schema-salad/default.nix
@@ -32,9 +32,8 @@ buildPythonPackage rec {
   ];
 
   checkInputs = [
-    black
     pytestCheckHook
-  ];
+  ] ++ passthru.optional-dependencies.pycodegen;
 
   disabledTests = [
     # Setup for these tests requires network access
@@ -48,6 +47,10 @@ buildPythonPackage rec {
     "schema_salad"
   ];
 
+  passthru.optional-dependencies = {
+    pycodegen = [ black ];
+  };
+
   meta = with lib; {
     description = "Semantic Annotations for Linked Avro Data";
     homepage = "https://github.com/common-workflow-language/schema_salad";