diff --git a/setup.py b/setup.py
index 33cae5a..d460638 100644
--- a/setup.py
+++ b/setup.py
@@ -1,12 +1,13 @@
 #!/usr/bin/env python3
 import setuptools
+import torch_agent
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
 setuptools.setup(
     name="torch-agent",
-    version="0.0.1",
+    version=torch_agent.__version__,
     author="B.J. Dweck",
     author_email="bjdweck@gmail.com",
     description="TORch: Iluminate the Way to your Node",