Update test_time_helper.py