Fix test failure caused by pylint corrections.