make python2 compatible