Undo header fix

I accidentally broke something the first time around