Fix for py 3