Forgot to uncomment import pyfits