Reverted back to testng assertions
Reverted back to testng assertions