Fix time-sensitive test