Fix astropy cosmology import.