Fix pylint tests