diff --git a/python/hccd/path_tracker.py b/python/hccd/path_tracker.py index 1013ad3..f42a98f 100644 --- a/python/hccd/path_tracker.py +++ b/python/hccd/path_tracker.py @@ -29,7 +29,7 @@ class PathTracker: def step(self, y): """Perform one predictor-corrector step.""" - return self.transparent_step(y)[0] + return self.transparent_step(y)[3] def transparent_step(self, y) -> typing.Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray,]: """Perform one predictor-corrector step, returning intermediate results."""