321. A person who receives a payment under a plan contemplated in section 318 or 320 pursuant to the death of the taxpayer must include in computing his income for the year only that part of such payment which would have been included under this chapter in computing the income of such taxpayer if he had received such amount under the plan.