Changed the tests from JUnit to TestNG