fixed JUnit import