Junit test case fix