fix for python 3

fix for python 3