1 files changed, 2 insertions, 1 deletions
diff --git a/src/main.rs b/src/main.rs
index 38c35e0..20ec270 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,6 +5,7 @@ use stellar_rs::event::event::Event;
 use stellar_rs::event::manager::EventHandle;
 use stellar_rs::event::manager::EventManager;
 use stellar_rs::event::manager::EventQueue;
+use stellar_rs::get_innermost_special_encapsulation;
 use stellar_rs::packet::capture::PacketCapture;
 use stellar_rs::packet::packet::Encapsulation;
 use stellar_rs::packet::packet::Packet;
@@ -141,7 +142,7 @@ fn handle_one_packet(mut packet: Packet, thread_ctx: &mut ThreadContext) {
         }
     }
 
-    match packet.get_inner_most_tuple() {
+    match get_innermost_special_encapsulation!(packet, TCP | UDP) {
         Some(_) => {
             let flow_id = packet.get_flow_id().unwrap();
             let session = session_mgr.update(flow_id);