Minor fix for py3