Fixed error in junit test