Add JUnit 4 back