State tax filing

The penalty is a federal thing, so make sure that you have done the federal part correctly, that it has been submitted and accepted. Once that is accepted you can go in and do what I suggested, because it basically makes the form feed in to the state calculations. Yes, you will have to remember the numbers, but you should also have a 1099 from the retirement account that lists the whole dollar amount and it will just be a third, or whatever amount was taxed this year. 

Of course all of the this gets more complicated if you plan on repaying these funds, and I don’t know yet how those calculations work.