diff --git a/massive/__init__.py b/massive/__init__.py
index da6c5136..3e662352 100644
--- a/massive/__init__.py
+++ b/massive/__init__.py
@@ -1,6 +1,6 @@
 from .rest import RESTClient
-from .rest.base import version
+from .rest.base import version, version_number
 from .websocket import WebSocketClient
 from .exceptions import *
 
-__version__ = version
+__version__ = version_number
diff --git a/massive/rest/base.py b/massive/rest/base.py
index 294dd482..8e2c4ac7 100644
--- a/massive/rest/base.py
+++ b/massive/rest/base.py
@@ -16,7 +16,7 @@
 logger = get_logger("RESTClient")
 version_number = "unknown"
 try:
-    version_number = version("massive-api-client")
+    version_number = version("massive")
 except PackageNotFoundError:
     pass
 
diff --git a/test_rest/test_version.py b/test_rest/test_version.py
new file mode 100644
index 00000000..abe80d5a
--- /dev/null
+++ b/test_rest/test_version.py
@@ -0,0 +1,20 @@
+import unittest
+from importlib.metadata import version
+
+import massive
+from massive import RESTClient
+
+
+class VersionTest(unittest.TestCase):
+    def test_version(self):
+        self.assertEqual(massive.__version__, version("massive"))
+
+    def test_user_agent(self):
+        self.assertEqual(
+            RESTClient(api_key="key").headers["User-Agent"],
+            f"Massive.com PythonClient/{version('massive')}",
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()