Fixed for python 3