Fixed the timing issue in the tests