Fix importing of astropy python module