Fix tooooooooo long wait time on unit test