Fix python3 issue