cleanup from pylint