Fixed time_tests.py