Update source to fix error. Add JUnit 4 to the classpath.