diff --git a/pkgs/development/python-modules/diagrams/default.nix b/pkgs/development/python-modules/diagrams/default.nix
index d34fc3b899f5..e015a6522d6e 100644
--- a/pkgs/development/python-modules/diagrams/default.nix
+++ b/pkgs/development/python-modules/diagrams/default.nix
@@ -27,7 +27,8 @@ buildPythonPackage rec {
 
   postPatch = ''
     substituteInPlace pyproject.toml \
-      --replace 'jinja2 = "^2.10"' 'jinja2 = "*"'
+      --replace 'jinja2 = "^2.10"' 'jinja2 = "*"' \
+      --replace 'graphviz = ">=0.13.2,<0.17.0"' 'graphviz = "*"'
   '';
 
   preConfigure = ''
@@ -35,7 +36,10 @@ buildPythonPackage rec {
     ./autogen.sh
   '';
 
-  patches = [ ./build_poetry.patch ];
+  patches = [
+    # The build-system section is missing
+    ./build_poetry.patch
+  ];
 
   checkInputs = [ pytestCheckHook ];
 
@@ -45,6 +49,8 @@ buildPythonPackage rec {
 
   propagatedBuildInputs = [ graphviz ];
 
+  pythonImportsCheck = [ "diagrams" ];
+
   meta = with lib; {
     description = "Diagram as Code";
     homepage    = "https://diagrams.mingrammer.com/";