diff --git a/distiller/__init__.py b/distiller/__init__.py
index 7947f31cac8cd78542080a2c1c92bf279aa9fdb9..f3367b1aa002503656fc9afea937649e383624f3 100755
--- a/distiller/__init__.py
+++ b/distiller/__init__.py
@@ -31,7 +31,7 @@ del dict_config
 del thinning
 
 # Distiller version
-__version__ = "0.3.1"
+__version__ = "0.4.0-pre"
 
 
 def model_find_param_name(model, param_to_find):