Fix for python 3 support