diff --git a/bec_widgets/cli/client.py b/bec_widgets/cli/client.py
index 7df48c90..c66dbac6 100644
--- a/bec_widgets/cli/client.py
+++ b/bec_widgets/cli/client.py
@@ -6,6 +6,7 @@ from __future__ import annotations
 import enum
 import inspect
 import traceback
+from functools import reduce
 from typing import Literal, Optional
 
 from bec_lib.logger import bec_logger
diff --git a/bec_widgets/cli/generate_cli.py b/bec_widgets/cli/generate_cli.py
index 9e9954ec..d98e23e4 100644
--- a/bec_widgets/cli/generate_cli.py
+++ b/bec_widgets/cli/generate_cli.py
@@ -40,6 +40,7 @@ class ClientGenerator:
             """import enum
 import inspect
 import traceback
+from functools import reduce
 from typing import Literal, Optional
 """
             if self._base
diff --git a/tests/unit_tests/test_generate_cli_client.py b/tests/unit_tests/test_generate_cli_client.py
index 2f76ea3a..83f29fde 100644
--- a/tests/unit_tests/test_generate_cli_client.py
+++ b/tests/unit_tests/test_generate_cli_client.py
@@ -73,6 +73,7 @@ def test_client_generator_with_black_formatting():
         import enum
         import inspect
         import traceback
+        from functools import reduce
         from typing import Literal, Optional
 
         from bec_lib.logger import bec_logger