Make setup script work on Python 3.