Fixed compatibility with JUnit