Issue about compatibility

Correct the compatibility about python3 and python2