Fix python3 compatability