Remove the setTimeout in the tests