Junit test modificatio.