Fixed pylint issue