diff --git a/Dockerfile b/Dockerfile
index 9933820..ad611a5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,6 @@
 FROM python:3.9
 WORKDIR /ramen
 RUN git clone https://github.com/zangobot/secml_malware
-RUN pip git+https://github.com/zangobot/ember.git
 RUN pip install -r /ramen/secml_malware/requirements.txt
 RUN pip install torch torchvision
 RUN pip install -e /ramen/secml_malware
diff --git a/requirements.txt b/requirements.txt
index ff57f8c..0788b16 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,5 @@ deap
 git+https://github.com/zangobot/ember.git
 requests
 torch
-torchvision
\ No newline at end of file
+torchvision
+tqdm
\ No newline at end of file