FIxed the test duration implementation