In the case of the death of either spouse, as of the beginning of the first taxable year of the spouse who survives following the taxable year in which such death occurred; except that if the spouse who survives is a citizen or resident of the United States who is a surviving spouse entitled to the benefits of section
2, the time provided by this subparagraph shall be as of the close of the last taxable year for which such individual is entitled to the benefits of section
2.