improve Junit test