Quick fix for Python 3 compatibility