Unless you meet one of the safe harbors, you can't just wait and pay a large balance due with tax return without penalty. If you don't have a sufficient amount of taxes withheld from the distribution, you might have to make an estimated tax payment to avoid penalties for quarterly underpayment of taxes:
https://www.irs.gov/taxtopics/tc306
