Fix setup on Python 3