Python 3 compatibility issue