Bug fix for Python 3.x compatibility