Fix python env setup.