You would enter a code PJ 2021 Form 1099-R as if you had already received it as I described in my reply to your first post. You would know from the amount requested to be returned and the gain-adjusted amount distributed what amounts go in boxes 1 and 2a.
