Small fix for Python 3