Remove JUnit dependency