Fix pylint complaint