diff --git a/docs/conf.py b/docs/conf.py
index 9998300..12e81a7 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -23,7 +23,7 @@
 author = 'Luca Demetrio'
 
 # The full version, including alpha/beta/rc tags
-release = '0.3.3'
+release = '0.4'
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/setup.py b/setup.py
index 075f6d3..0a99e25 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 with open(str(current_dir / "README.md"), "r") as f:
     long_description = f.read()
 
-version = '0.3.2'
+version = '0.4'
 
 setup(
     name="secml_malware",
@@ -25,7 +25,7 @@
         'deap',
         'torch',
         'torchvision',
-        'tqdm',
+        'git+https://github.com/zangobot/ember.git'
     ],
     url="https://github.com/pralab/secml_malware",
     license="GPL 3.0",