Make it python2 compatible