Fix for older astropy modules