Import astropy correctly
Fix imports