Made it JUnit compatible