Fix pylint failures in CI