Changed the return with units method.

In response to comment here:
https://github.com/astropy/astropy/pull/2103#commitcomment-5470863