Fix for Python 3 compatibility