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