diff --git a/Dockerfile b/Dockerfile
index 9b3a6da6..5ca68273 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -35,7 +35,7 @@ COPY ./lit_nlp/examples/gunicorn_config.py ./
 
 FROM lit-nlp-base AS lit-nlp-prod
 
-RUN python -m pip install 'lit-nlp[examples-descai]'
+RUN python -m pip install 'lit-nlp[examples-discriminative-ai]'
 
 WORKDIR $APP_HOME
 ENTRYPOINT ["gunicorn", "--config=gunicorn_config.py"]
diff --git a/lit_nlp/examples/gcp/Dockerfile b/lit_nlp/examples/gcp/Dockerfile
index 707ed6a4..86217e1d 100644
--- a/lit_nlp/examples/gcp/Dockerfile
+++ b/lit_nlp/examples/gcp/Dockerfile
@@ -93,14 +93,14 @@ COPY ./lit_nlp/examples/gcp/model_server_gunicorn_config.py ./
 
 FROM lit-gcp-app-server-base AS lit-gcp-app-server
 
-RUN python -m pip install 'lit-nlp[examples-genai]'
+RUN python -m pip install 'lit-nlp[examples-generative-ai]'
 ENTRYPOINT ["gunicorn", "--config=gunicorn_config.py"]
 
 
 
 FROM lit-gcp-model-server-base AS lit-gcp-model-server
 
-RUN python -m pip install 'lit-nlp[examples-genai]'
+RUN python -m pip install 'lit-nlp[examples-generative-ai]'
 ENTRYPOINT ["gunicorn", "--config=model_server_gunicorn_config.py"]