Make env-setup work on Python 3