diff --git a/pkgs/by-name/do/docling-serve/package.nix b/pkgs/by-name/do/docling-serve/package.nix
index df0fb39a4724..ff8e4514d60e 100644
--- a/pkgs/by-name/do/docling-serve/package.nix
+++ b/pkgs/by-name/do/docling-serve/package.nix
@@ -1,6 +1,10 @@
-{ python3Packages, nixosTests }:
+{
+  python3Packages,
+  nixosTests,
+  withUI ? false,
+}:
 
-(python3Packages.toPythonApplication python3Packages.docling-serve)
+(python3Packages.toPythonApplication (python3Packages.docling-serve.override { inherit withUI; }))
 // {
   passthru.tests = {
     docling-serve = nixosTests.docling-serve;