fix for py2