Fix compatibility with Astropy 0.3