Fix the python