Fixed test error in Python 3.