diff --git a/pkgs/by-name/di/diagrams-as-code/package.nix b/pkgs/by-name/di/diagrams-as-code/package.nix
index cbce445d630a919..fd748332c230eab 100644
--- a/pkgs/by-name/di/diagrams-as-code/package.nix
+++ b/pkgs/by-name/di/diagrams-as-code/package.nix
@@ -29,6 +29,7 @@ python3Packages.buildPythonPackage rec {
   pythonRelaxDeps = [
     "diagrams"
     "pydantic"
+    "pyyaml"
   ];
 
   pythonImportsCheck = [ "diagrams_as_code" ];