improve Junit