fix a bug and add two test, next finish JUnit first