No - you don't have to amend your 2016 return since you haven't received the payout yet.
Normally the plan administrator will withhold federal taxes on the payment. If they don't you should make an estimated tax payment after you receive the funds. You might also want to make an estimated payment on your state taxes.
**Say "Thanks" by clicking the thumb icon in a post
**Mark the post that answers your question by clicking on "Mark as Best Answer"