diff --git a/bec_widgets/cli/client.py b/bec_widgets/cli/client.py
index 11a5e94b..5fe5dfe3 100644
--- a/bec_widgets/cli/client.py
+++ b/bec_widgets/cli/client.py
@@ -7,6 +7,7 @@ import enum
 import inspect
 import traceback
 from functools import reduce
+from operator import add
 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 f53f5a04..fab9a803 100644
--- a/bec_widgets/cli/generate_cli.py
+++ b/bec_widgets/cli/generate_cli.py
@@ -41,6 +41,7 @@ class ClientGenerator:
 import inspect
 import traceback
 from functools import reduce
+from operator import add
 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 83f29fde..2ecf3915 100644
--- a/tests/unit_tests/test_generate_cli_client.py
+++ b/tests/unit_tests/test_generate_cli_client.py
@@ -74,6 +74,7 @@ def test_client_generator_with_black_formatting():
         import inspect
         import traceback
         from functools import reduce
+        from operator import add
         from typing import Literal, Optional
 
         from bec_lib.logger import bec_logger