Retirement tax questions

You must claim what you received in 2016 on your 2016 tax return.

If you repay it in 2017, the repayment will be addressed on your 2017 return (filed in early 2018).