Fix an import within `ode_system.py`