Fix junit test error