Fix broken pylint test