junit test correction