Minor fixes due to change in Astropy API