Fixed python 3 syntax