Corrected JUnit file