Update morphology imports for astropy 0.4