Fix broken junit