Fix for python 3 too