Fix test and pylint.