Fix pep error