diff --git a/docker/1.3.1/py3/Dockerfile.eia b/docker/1.3.1/py3/Dockerfile.eia
index 5f7fc680..1c040c97 100644
--- a/docker/1.3.1/py3/Dockerfile.eia
+++ b/docker/1.3.1/py3/Dockerfile.eia
@@ -67,7 +67,6 @@ RUN conda install -c \
  && conda install -y \
     scikit-learn==0.21.2 \
     pandas==0.25.0 \
-    pillow==6.2.1 \
     h5py==2.9.0 \
     requests==2.22.0 \
  && conda clean -ya \
@@ -78,7 +77,8 @@ RUN conda install -c \
  && pip install https://download.pytorch.org/whl/cpu/torchvision-0.4.2%2Bcpu-cp36-cp36m-linux_x86_64.whl \
  && pip install https://s3.amazonaws.com/amazonei-pytorch/torch_eia-1.3.1-cp36-cp36m-manylinux1_x86_64.whl \
  && pip install graphviz==$GRAPHVIZ_VERSION \
- && pip install mxnet-model-server==$MMS_VERSION
+ && pip install mxnet-model-server==$MMS_VERSION \
+ && pip install pillow==7.1.0
 
 RUN useradd -m model-server \
  && mkdir -p /home/model-server/tmp \
@@ -93,6 +93,9 @@ RUN pip install --no-cache-dir "sagemaker-pytorch-inference<2"
 
 RUN curl https://aws-dlc-licenses.s3.amazonaws.com/pytorch/license.txt -o /license.txt
 
+RUN conda install -y -c conda-forge pyyaml==5.3.1 \
+ && pip install sagemaker-containers==2.8.6
+
 RUN wget https://amazonei-tools.s3.amazonaws.com/v${HEALTH_CHECK_VERSION}/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz -O /opt/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz \
  && tar -xvf /opt/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz -C /opt/ \
  && rm -rf /opt/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz \