Fixed import line in time_tests.py