diff --git a/docker-cpu.df b/docker-cpu.df
index dcda40a..d9770ba 100644
--- a/docker-cpu.df
+++ b/docker-cpu.df
@@ -20,7 +20,7 @@ RUN wget --quiet https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86
 WORKDIR /ne
 COPY requirements.txt .
 RUN /opt/conda/bin/conda install -q -y conda numpy scipy pip pillow
-RUN /opt/conda/bin/python3.5 -m pip install -q -r "requirements.txt"
+RUN /opt/conda/bin/python3 -m pip install -q -r "requirements.txt"
 
 # Copy only required project files
 COPY enhance.py .
diff --git a/docker-gpu.df b/docker-gpu.df
index 788a4d9..aba05a6 100644
--- a/docker-gpu.df
+++ b/docker-gpu.df
@@ -18,7 +18,7 @@ RUN wget --quiet https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86
 WORKDIR /ne
 COPY requirements.txt .
 RUN /opt/conda/bin/conda install -q -y conda numpy scipy pip pillow
-RUN /opt/conda/bin/python3.5 -m pip install -q -r "requirements.txt"
+RUN /opt/conda/bin/python3 -m pip install -q -r "requirements.txt"
 
 # Copy only required project files
 COPY enhance.py .