Fix python3 error