Fix installation on Python 2