Fixed tests in test_ode.py