Add lb into proto