"Just need to go back and amend to show the repayment"
You don't really show a repayment. For a rollover, you mark the 1099-R worksheet that it was rolled over.
