Change pyfits for astropy.io.fits