Include correct NAXIS in header

The NAXIS keywords need to be updated in the header (WCS doesn't know about NAXIS)