some changes of Junit test