Fixed python2 bug