diff --git a/stack/base/Dockerfile b/stack/base/Dockerfile
index 53a6dc31..6f035fad 100644
--- a/stack/base/Dockerfile
+++ b/stack/base/Dockerfile
@@ -52,6 +52,7 @@ RUN mamba update -y pip async_generator certipy && \
      mamba install --yes \
      aiida-core==${AIIDA_VERSION} \
      mamba-bash-completion \
+     traitlets=5.9.0 \
      && mamba clean --all -f -y && \
      fix-permissions "${CONDA_DIR}"