Looks good.  Note that there is already a variable, `PY3`, defined at the top of this file for this purpose.  (We don't generally have that in astropy, but `setup_helpers.py` is a special file that needs to be imported without having `2to3` run on it first).  So you could change this line to simply:

```
   if PY3:
```