Switched to JUnit, hope it doesn't mess up