Fix tests
pylint