Yes, you can help avoid underpayment penalties by adjusting your W-4 to increase state withholding.
You can also contact the plan administrators for your 401K and/or your pension and ask them to withhold state taxes. This might be a better solution if you are going to continue getting the retirement income but may not keep your "post-retirement" job.
