I'll assume that these were contributions reported with codes D and AA on a W-2. The only portion that is to be added to 2021 income is the portion returned from the traditional 401(k) account, originally reported with code D on the W-2. This needs to be added to 2021 income because it was not included in the amount in box 1 of the W-2. (The amount contributed as Roth contributions was included in box 1 and therefore has already been included in income.)
