Fix for Python 3.x