Fixing a py3 issue