Fixed python 3 issue