diff --git a/VERSION b/VERSION
index 6812f812..05b19b1f 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-0.0.3
\ No newline at end of file
+0.0.4
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 7f5bd159..cb20fb1e 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@
     ],
     entry_points={
         "console_scripts": [
-            "pandora = pandora.__main__:main",
+            "pandora = pandora.__main__:run",
         ]
     }
 )
diff --git a/src/__init__.py b/src/__init__.py
index fc02d3ce..eae45660 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -1,3 +1,3 @@
 # -*- coding: utf-8 -*-
 
-__version__ = '0.0.3'
+__version__ = '0.0.4'
diff --git a/src/__main__.py b/src/__main__.py
index 827a6307..9e4dc153 100644
--- a/src/__main__.py
+++ b/src/__main__.py
@@ -48,7 +48,7 @@ def main():
     ChatBot(ChatGPT(access_token, args.proxy)).run()
 
 
-if __name__ == '__main__':
+def run():
     try:
         main()
     except KeyboardInterrupt:
@@ -56,3 +56,7 @@ def main():
         sys.exit(0)
     except Exception as e:
         Console.error_bh('### Error occurred: ' + str(e))
+
+
+if __name__ == '__main__':
+    run()