Changing assertEquals from JUnit to TestNG within TestNG-example