Convert to testNG