From cf93155278075cc760ffe10b60539c62a15dccbc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= <fabian@schaluck.com>
Date: Thu, 5 Dec 2024 23:44:06 +0000
Subject: [PATCH] fix template

---
 python/sdist/amici/jax/jax.template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/sdist/amici/jax/jax.template.py b/python/sdist/amici/jax/jax.template.py
index d395715422..5d5521d222 100644
--- a/python/sdist/amici/jax/jax.template.py
+++ b/python/sdist/amici/jax/jax.template.py
@@ -97,7 +97,7 @@ def observable_ids(self):
 
     @property
     def state_ids(self):
-        return TPL_X_IDS
+        return TPL_X_RDATA_IDS
 
     @property
     def parameter_ids(self):