Python 2 fixes