Small pylint fixes