Fix for py2