diff --git a/pkgs/development/python-modules/dtschema/default.nix b/pkgs/development/python-modules/dtschema/default.nix
index 377c315d9d92..cb043872a82c 100644
--- a/pkgs/development/python-modules/dtschema/default.nix
+++ b/pkgs/development/python-modules/dtschema/default.nix
@@ -29,6 +29,10 @@ buildPythonPackage rec {
     rfc3987
   ];
 
+  pythonImportsCheck = [
+    "dtschema"
+  ];
+
   meta = with lib; {
     description = "Tooling for devicetree validation using YAML and jsonschema";
     homepage = "https://github.com/devicetree-org/dt-schema/";