Fix pylint issue