update test to use JUnit 4 correctly