Pylint fixes

Hope the build will be green