From 79d24ea3cf800cddc526f5d6bd3b3167e892f2f4 Mon Sep 17 00:00:00 2001
From: Francesco Innocenti <F.Innocenti@sussex.ac.uk>
Date: Mon, 24 Jun 2024 13:25:42 +0100
Subject: [PATCH] Fix small bug in activity grad

---
 jpc/_core/_grads.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/jpc/_core/_grads.py b/jpc/_core/_grads.py
index 893af70..7c77d1d 100644
--- a/jpc/_core/_grads.py
+++ b/jpc/_core/_grads.py
@@ -39,7 +39,7 @@ def _neg_activity_grad(
 
     """
     generator, output, input = args
-    dFdzs = grad(pc_energy_fn, argnums=1)(
+    dFdzs = grad(energy_fn, argnums=1)(
         generator,
         activities,
         output,