Fix python version issues