After you file

when your federal amended return goes through,

the $400 in your example will be correct.