diff --git a/mlops-template-gitlab/seedcode/mlops-gitlab-project-seedcode-model-build/.gitlab-ci.yml b/mlops-template-gitlab/seedcode/mlops-gitlab-project-seedcode-model-build/.gitlab-ci.yml
index 1ef66107..044c9387 100644
--- a/mlops-template-gitlab/seedcode/mlops-gitlab-project-seedcode-model-build/.gitlab-ci.yml
+++ b/mlops-template-gitlab/seedcode/mlops-gitlab-project-seedcode-model-build/.gitlab-ci.yml
@@ -21,6 +21,7 @@ train-model:
   stage: training
   script: 
       - pip install --upgrade --force-reinstall . "awscli>1.20.30"
+      - pip install --upgrade --force-reinstall . setuptools
       - export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
       - export PYTHONUNBUFFERED=TRUE
       - export SAGEMAKER_PROJECT_NAME_ID="${SAGEMAKER_PROJECT_NAME}-${SAGEMAKER_PROJECT_ID}"