Story 3: fix the test failing due time mismatch