Fixed python 3 error