diff --git a/Dockerfile.lmes-job b/Dockerfile.lmes-job
index e6268a31..31efe6c8 100644
--- a/Dockerfile.lmes-job
+++ b/Dockerfile.lmes-job
@@ -15,7 +15,7 @@ ENV PATH="/opt/app-root/bin:/opt/app-root/src/.local/bin/:/opt/app-root/src/bin:
 # Clone the Git repository, check out v0.4.4 and install the Python package
 RUN git clone https://github.com/opendatahub-io/lm-evaluation-harness.git && \
     cd lm-evaluation-harness &&  git checkout release-0.4.5 && \
-    pip install --no-cache-dir --user -e .[api]
+    pip install --no-cache-dir --user -e .[api,ibm_watsonx_ai]
 
 RUN python -c 'from lm_eval.tasks.unitxt import task; import os.path; print("class: !function " + task.__file__.replace("task.py", "task.Unitxt"))' > ./my_tasks/unitxt