Fix Junit issue