收到前雇主的401K的 improper distribution 如何报税

去年跳槽后，可能是因为没有及时通知HR，导致工资发多了，后来又退回去了，其中包括退回的 Roth 401K。但是现在报税的时候，401K的数值好像不对，求问改如何处理？

大致时间线如下:

T1:  离职
T2:  收到最后一笔工资，包括401K
T3:  收到前雇主的信件，说工资发多了 S，需要退回去
T4:  退回多发的工资  S
T5:  等了一个多月后没见到修正 401K 的信件，以为没事了，就把前雇主的 401K 转到了新雇主那里
T6:  收到前雇主通过 Fidelity发来的信件，说401K 由于 administrative error 发多了 X, 属于 Improper Distribution, 需要 repay X。 Repay 的方式如下

1. Authorizing your IRA custodian (if you rolled the distribution over to an IRA) to deduct the amount of Improper Distribution from your IRA and recontributing it to the Plan; or
2. Submitting a personal or certified check in the amount of the Improper Distribution to the Plan

T7:  由于楼主是rollover到了新雇主的401K，所以选择了方案2， 寄了一张 checking account 的支票给 Fidelity
T8: 今年收到税表后，发现前雇主 W2 表里面 Roth 401K (box 12a 里面的 AA) 的contribute 是 Y. 并且box 1 里面的income 是 T. 楼主将以上数据和 paystub 对比了一下发现:

Paystub 里的 income = T + S
Paystub 里的 Roth 401K contribution = Y + X

所有的数字看起来是对的，但是前雇主因为多放了 X 到 401K 里面，我repay的时候是从自己的 checking account 里面 repay 的，所以看起似乎是我间接地自己往 Roth 401K 里面存了 X, 并且这 X 没有反映在我的 W2 表上。这 X 虽然已经是税后的，但是因为是在 Roth 401K里面，所以相当于 capital gain 不需要交税；但是这显然是有问题的。楼主其他能找到的税表里面，好像都不可能算出这个 X 值。所以求问这种情况该怎么处理？

另外，楼主在新公司的 401K contribution 是 Z, 而 Z + Y 本身已经超过了 $20500 的上限。所以我现在应该退回 Z + Y - $20500 还是 Z + Y + X - $20500?