diff --git a/.github/workflows/pythonpackage.yml b/.github/workflows/pythonpackage.yml
index d37c116..c94cc5b 100644
--- a/.github/workflows/pythonpackage.yml
+++ b/.github/workflows/pythonpackage.yml
@@ -15,7 +15,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.8, 3.9]
+        python-version: [3.11, 3.12, 3.13]
 
     steps:
     - uses: actions/checkout@v2
diff --git a/jax_cosmo/__init__.py b/jax_cosmo/__init__.py
index b89ac54..81e7868 100644
--- a/jax_cosmo/__init__.py
+++ b/jax_cosmo/__init__.py
@@ -1,9 +1,9 @@
 # Cosmology in JAX
-from pkg_resources import DistributionNotFound, get_distribution
+from importlib.metadata import PackageNotFoundError, version
 
 try:
-    __version__ = get_distribution(__name__).version
-except DistributionNotFound:
+    __version__ = version(__name__)
+except PackageNotFoundError:
     # package is not installed
     pass