Fix python3 problem