diff --git a/jaxutils/__init__.py b/jaxutils/__init__.py
index 3fa7a54..b166dcc 100644
--- a/jaxutils/__init__.py
+++ b/jaxutils/__init__.py
@@ -16,9 +16,7 @@
 
 from .pytree import PyTree
 from .data import Dataset, verify_dataset
-from . import _version
 
-__version__ = _version.get_versions()["version"]
 __authors__ = "Thomas Pinder, Daniel Dodd"
 __license__ = "MIT"
 __emails__ = "tompinder@live.co.uk, d.dodd1@lancaster.ac.uk"
@@ -35,3 +33,7 @@
     "Dataset",
     "verify_dataset",
 ]
+
+from . import _version
+
+__version__ = _version.get_versions()["version"]
diff --git a/setup.cfg b/setup.cfg
index 082e4bf..c92efe6 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -3,7 +3,7 @@ VCS = git
 style = pep440
 versionfile_source = jaxutils/_version.py
 versionfile_build = jaxutils/_version.py
-tag_prefix = ''
+tag_prefix = v
 
 [flake8]
 exclude =
diff --git a/setup.py b/setup.py
index e6a1877..ef41478 100644
--- a/setup.py
+++ b/setup.py
@@ -22,9 +22,9 @@
 
 
 # Handle builds of nightly release - adapted from BlackJax.
-if "BUILD_JAXUTILS_NIGHTLY" in os.environ:
-    if os.environ["BUILD_JAXUTILS_NIGHTLY"] == "nightly":
-        NAME += "-nightly"
+# if "BUILD_JAXUTILS_NIGHTLY" in os.environ:
+#     if os.environ["BUILD_JAXUTILS_NIGHTLY"] == "nightly":
+#         NAME += "-nightly"
 
 REQUIRES = ["jax>=0.4.0", "jaxlib>=0.4.0", "jaxtyping"]
 
@@ -42,7 +42,7 @@
 
 
 setup(
-    name="JaxUtils",
+    name=NAME,
     version=versioneer.get_version(),
     cmdclass=versioneer.get_cmdclass(),
     author="Daniel Dodd and Thomas Pinder",