diff --git a/setup.py b/setup.py
index 139fec9..6a75c06 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
 # This call to setup() does all the work
 setup(
     name="torch_choice",
-    version="0.0.3",
+    version="0.0.4",
     description="A Pytorch Backend Library for Choice Modelling",
     long_description=README,
     long_description_content_type="text/markdown",