Reverted back to testng assertions