Fix JUnit test cases
Fix JUnit test cases