Fixed pylint error