diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index fe05a51..b341925 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -41,7 +41,7 @@ jobs:
         run: python -c "import sys; print(sys.version)"
       - name: Install sdist without optional dependencies
         run: pip install dist/*.tar.gz
-      - run: python -c 'import trx.version; print(trx.version.__version__)'
+      - run: python -c 'import trx; print(trx.__version__)'
       - name: Install pytest
         run: pip install pytest psutil pytest-console-scripts pytest-cov
       - name: Run tests
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index e9f231e..40e83d1 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -34,6 +34,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           python -m pip install -e .[dev,test]
+          python -c "import trx; print(trx.__version__)"
 
       - name: Test
         run: spin test
diff --git a/pyproject.toml b/pyproject.toml
index 18de8f9..2f6f606 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -86,7 +86,7 @@ packages = ["trx"]
 include-package-data = true
 
 [tool.setuptools_scm]
-write_to = "trx/version.py"
+write_to = "trx/_version.py"
 fallback_version = "0.0"
 local_scheme = "no-local-version"
 
diff --git a/trx/__init__.py b/trx/__init__.py
index 047e068..1559ad5 100644
--- a/trx/__init__.py
+++ b/trx/__init__.py
@@ -1 +1,11 @@
 """TRX file format for brain tractography data."""
+
+try:
+    from ._version import __version__  # noqa: F401
+except ImportError:
+    try:
+        from importlib.metadata import PackageNotFoundError, version
+
+        __version__ = version("trx-python")
+    except (ImportError, PackageNotFoundError):
+        __version__ = "unknown"