diff --git a/container/build/Dockerfile b/container/build/Dockerfile
index 8b6768c190..407432c106 100644
--- a/container/build/Dockerfile
+++ b/container/build/Dockerfile
@@ -18,6 +18,6 @@ COPY requirements.txt /opt/
 RUN /opt/python/cp310-cp310/bin/pip install -r /opt/requirements.txt && \
     /opt/python/cp311-cp311/bin/pip install -r /opt/requirements.txt && \
     pipx install isort==5.12.0 --python python3.11 && \
-    pipx install black==23.7.0 --python python3.11
+    pipx install black==23.7.0 --python python3.11 
 
 RUN git config --global --add safe.directory '*'
diff --git a/requirements.txt b/requirements.txt
index 7f7fed8441..828e917a7f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,4 +15,5 @@ pandas~=2.1
 # ci deps
 jinja2
 black==23.7.0
+jupyter-black==0.3.4
 isort==5.12.0