Corrected junit test