diff --git a/Dockerfile b/Dockerfile
index 32f5639..5b01c40 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,7 +8,7 @@ RUN apt-get update && apt-get install -y python3 python3-pip
 WORKDIR /home/watcloud-slurm-ci/
 
 # Copy the Python script and any necessary files
-COPY main.py config.py runner_size_config.py RunningJob.py allocation_scripts/apptainer.sh start.sh /home/watcloud-slurm-ci/
+COPY main.py config.py runner_size_config.py RunningJob.py KubernetesLogFormatter.py allocation_scripts/apptainer.sh start.sh /home/watcloud-slurm-ci/
 
 # Install Python requirements
 COPY requirements.txt /home/watcloud-slurm-ci/