fix: Python 3 errors