Fixed test on Python 3