small Python 2 fix