401k refund after plan failed discrimination test

I have already filed my taxes for 2020, but then later in March, I received a refund from my 401k because I was told my company's 401K plan failed the discrimination test. My question is: should I file an amendment for 2020 (because this was for the 2020 retirement contribution), or should I report this on 2021 (because this was distributed to me in 2021, and the 1099 will not be generated until 2021)? 

 

I'm getting conflicting answers. My plan administration is saying I should file it for 2020, but when I called the 401K brokerage company, they tell me I should file it for 2021.