Revert changes for JUnit dependency