diff --git a/info.py b/info.py
index 9f2f0a97..2eb2a82e 100644
--- a/info.py
+++ b/info.py
@@ -1,8 +1,9 @@
 from collections import OrderedDict
+from version import VERSION
 
 v = OrderedDict()
 v["name"] = "ai-toolkit"
 v["repo"] = "https://github.com/ostris/ai-toolkit"
-v["version"] = "0.1.0"
+v["version"] = VERSION
 
 software_meta = v
diff --git a/version.py b/version.py
index b48ee15a..0db14ed3 100644
--- a/version.py
+++ b/version.py
@@ -1 +1 @@
-VERSION = "0.3.0"
\ No newline at end of file
+VERSION = "0.3.1"
\ No newline at end of file