fix for Python 3

small fix for Python 3