After you file

No matter when you got the stimulus payments  you DID get them  so enter the amount you did get on the return.