Quick Python 3 fix