Fixed python2 bug.