Fixed compatibility with Astropy 0.2