No, don't return your refund. If your 1040-NR shows a smaller refund that what you received, pay the difference with the amended return. If your 1040-NR shows a larger refund, the IRS should send you the additional refund.
