diff --git a/Dockerfile b/Dockerfile
index 70f1f27..6e58e1c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ubuntu:latest
+FROM ubuntu:20.04
 
 ENV PYTHONIOENCODING=utf-8
 
@@ -9,7 +9,8 @@ RUN pip3 install requests python-Levenshtein tqdm sacred pymongo mlflow
 RUN pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
 RUN apt install -y libgit2-dev
 RUN pip3 install setuptools-rust python-dev-tools
-RUN pip3 install dvc[ssh] paramiko
+RUN pip3 install dvc 'dvc[ssh]' paramiko
+
 # RUN apt install -y wget
 # RUN wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh
 # RUN chmod +x ./Anaconda3-2021.05-Linux-x86_64.sh