Temporary fix to use NAXIS1/2 now that astropy hides them