Correct pylint errors