Fixing the python