Fix py 3.x compatibility

I hope - haven't installed pythone 3 yet