Fix Python environment