diff --git a/brainpy/_src/analysis/lowdim/lowdim_analyzer.py b/brainpy/_src/analysis/lowdim/lowdim_analyzer.py
index 78ab3f44c..4303543b8 100644
--- a/brainpy/_src/analysis/lowdim/lowdim_analyzer.py
+++ b/brainpy/_src/analysis/lowdim/lowdim_analyzer.py
@@ -436,7 +436,7 @@ def F_fy(self):
       f = partial(f, **(self.pars_update + self.fixed_vars))
       f = utils.f_without_jaxarray_return(f)
       f = utils.remove_return_shape(f)
-      self.analyzed_results[C.F_fy] = bm.jit(f, device=self.jit_device)
+      self.analyzed_results[C.F_fy] = jax.jit(f, device=self.jit_device)
     return self.analyzed_results[C.F_fy]
 
   @property