Junit test bug fixed