Minor pylint fixs