Fix pylint error