Add assertion to check the update result