Final (hopefully) pylint fix