diff --git a/jpc/__init__.py b/jpc/__init__.py
index fea64ab..f81f798 100644
--- a/jpc/__init__.py
+++ b/jpc/__init__.py
@@ -17,4 +17,4 @@
 )
 
 
-__version__ = importlib.metadata.version("jpc_local")
+__version__ = importlib.metadata.version("jpc")
diff --git a/jpc/_grads.py b/jpc/_grads.py
index e5edb51..5f0fcd4 100644
--- a/jpc/_grads.py
+++ b/jpc/_grads.py
@@ -4,7 +4,7 @@
 from equinox import filter_grad
 from jaxtyping import PyTree, ArrayLike, Array
 from typing import Union, Tuple, Callable, Optional
-from _energies import _energy_fn, pc_energy_fn, _lateral_energy_fn
+from ._energies import _energy_fn, pc_energy_fn, _lateral_energy_fn
 
 
 def _neg_activity_grad(
diff --git a/jpc/_infer.py b/jpc/_infer.py
index 426fd2f..49cac0d 100644
--- a/jpc/_infer.py
+++ b/jpc/_infer.py
@@ -2,7 +2,7 @@
 
 from jaxtyping import PyTree, ArrayLike, Array
 from typing import Callable, Optional, Union
-from _grads import _neg_activity_grad, _neg_lateral_activity_grad
+from ._grads import _neg_activity_grad, _neg_lateral_activity_grad
 from diffrax import (
     AbstractSolver,
     AbstractStepSizeController,
diff --git a/pyproject.toml b/pyproject.toml
index ef506a7..dbc405e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,8 +3,8 @@ name = "jpc"
 version = "0.0.1"
 description = "Predictive coding networks in JAX."
 readme = "README.md"
-requires-python ="~=3.8"
-#license = {file = "LICENSE"}
+requires-python ="~=3.9"
+license = {file = "LICENSE"}
 authors = [
   {name = "Francesco Innocenti", email = "F.Innocenti@sussex.ac.uk"},
 ]
@@ -33,10 +33,10 @@ classifiers = [
 ]
 urls = {repository = "https://github.com/thebuckleylab/jpc"}
 dependencies = [
-    "jax>=0.4.13",
-    "equinox>=0.10.4",
-    "diffrax>=0.3.1",
-    "jaxtyping>=0.2.19"
+    "jax>=0.4.23",
+    "equinox>=0.11.2",
+    "diffrax>=0.5.1",
+    "jaxtyping>=0.2.24"
 ]
 
 [build-system]