diff --git a/core/benches/proto_to_packet.rs b/core/benches/proto_to_packet.rs index 87f85f9c7f..1727d8af9c 100644 --- a/core/benches/proto_to_packet.rs +++ b/core/benches/proto_to_packet.rs @@ -25,6 +25,7 @@ fn get_proto_packet(i: u8) -> PbPacket { repair: false, simple_vote_tx: false, tracer_packet: false, + from_staked_node: false, }), sender_stake: 0, }), diff --git a/core/src/lib.rs b/core/src/lib.rs index 5c4aa86523..9d6908b317 100644 --- a/core/src/lib.rs +++ b/core/src/lib.rs @@ -106,6 +106,12 @@ pub fn proto_packet_to_packet(p: jito_protos::proto::packet::Packet) -> Packet { if flags.repair { packet.meta_mut().flags.insert(PacketFlags::REPAIR); } + if flags.from_staked_node { + packet + .meta_mut() + .flags + .insert(PacketFlags::FROM_STAKED_NODE) + } } } packet