Fixed python 3 compatibility