diff --git a/LinSATNet/__init__.py b/LinSATNet/__init__.py
index b92614f..d6c0182 100644
--- a/LinSATNet/__init__.py
+++ b/LinSATNet/__init__.py
@@ -5,5 +5,5 @@
 # Code author: Runzhong Wang (runzhong.wang@outlook.com)
 
 from .linsat import linsat_layer
-__version__ = '0.0.3'
+__version__ = '0.0.4'
 __author__ = 'Runzhong Wang'