Small fix for time of testing