Handle JUnit 4 tests as well