small change and add Junit test