[setup] Fix setup for Python 3

Fixes #478