Couple of pylint fixes