Use TestNG assert instead of JUnit