After you file

Yes, you enter the stimulus payment that you received in January, 2021 on the 2020 income tax return.  The payment received in January, 2021 is retroactive to 2020.  

@legree76