Fix junit test