Remove sleeps from tests