Fix pylint part 1