Make compatible with pyfits > 3.0