Attempt to fix for Python 3