From 3eaaa8eac30b9603bac9e5af41621e7c9f09ebdf Mon Sep 17 00:00:00 2001
From: Heiko Zimmermann <zimmermann.ho@gmail.com>
Date: Wed, 23 Apr 2025 11:24:47 +0200
Subject: [PATCH] jvp should only return derivative

---
 demo/cfd/cfd-tesseract/tesseract_api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/cfd/cfd-tesseract/tesseract_api.py b/demo/cfd/cfd-tesseract/tesseract_api.py
index 5b3ab39..6d9caab 100644
--- a/demo/cfd/cfd-tesseract/tesseract_api.py
+++ b/demo/cfd/cfd-tesseract/tesseract_api.py
@@ -204,7 +204,7 @@ def jvp_jit(
         filtered_apply,
         [flatten_with_paths(inputs, include_paths=jvp_inputs)],
         [tangent_vector],
-    )
+    )[1]
 
 
 @eqx.filter_jit