Fixed Python 3 compatibility