Fix junit issue