updated assertions as per review comments