fix a problem in junit test