Fix setup and pylint error