1 files changed, 5 insertions, 5 deletions

diff --git a/src/lib/protocols/thunder.c b/src/lib/protocols/thunder.c
index 2f31672e7..0d905dfe8 100644
--- a/src/lib/protocols/thunder.c
+++ b/src/lib/protocols/thunder.c
@@ -33,7 +33,7 @@ static void ndpi_int_thunder_add_connection(struct ndpi_detection_module_struct
   struct ndpi_id_struct *src = flow->src;
   struct ndpi_id_struct *dst = flow->dst;
 
-  ndpi_int_add_connection(ndpi_struct, flow, NDPI_PROTOCOL_THUNDER/* , protocol_type */);
+  ndpi_set_detected_protocol(ndpi_struct, flow, NDPI_PROTOCOL_THUNDER);
 
   if (src != NULL) {
     src->thunder_ts = packet->tick_timestamp;
@@ -62,7 +62,7 @@ void ndpi_int_search_thunder_udp(struct ndpi_detection_module_struct
       && packet->payload[0] < 0x40 && packet->payload[1] == 0 && packet->payload[2] == 0 && packet->payload[3] == 0) {
     if (flow->thunder_stage == 3) {
       NDPI_LOG(NDPI_PROTOCOL_THUNDER, ndpi_struct, NDPI_LOG_DEBUG, "THUNDER udp detected\n");
-      ndpi_int_thunder_add_connection(ndpi_struct, flow/* , NDPI_REAL_PROTOCOL */);
+      ndpi_int_thunder_add_connection(ndpi_struct, flow);
       return;
     }
 
@@ -96,7 +96,7 @@ void ndpi_int_search_thunder_tcp(struct ndpi_detection_module_struct
       && packet->payload[0] < 0x40 && packet->payload[1] == 0 && packet->payload[2] == 0 && packet->payload[3] == 0) {
     if (flow->thunder_stage == 3) {
       NDPI_LOG(NDPI_PROTOCOL_THUNDER, ndpi_struct, NDPI_LOG_DEBUG, "THUNDER tcp detected\n");
-      ndpi_int_thunder_add_connection(ndpi_struct, flow/* , NDPI_REAL_PROTOCOL */);
+      ndpi_int_thunder_add_connection(ndpi_struct, flow);
       return;
     }
 
@@ -126,7 +126,7 @@ void ndpi_int_search_thunder_tcp(struct ndpi_detection_module_struct
 	&& packet->payload[packet->empty_line_position + 5] == 0x00) {
       NDPI_LOG(NDPI_PROTOCOL_THUNDER, ndpi_struct, NDPI_LOG_DEBUG,
 	       "maybe thunder http POST packet application does match\n");
-      ndpi_int_thunder_add_connection(ndpi_struct, flow/* , NDPI_CORRELATED_PROTOCOL */);
+      ndpi_int_thunder_add_connection(ndpi_struct, flow);
       return;
     }
   }
@@ -188,7 +188,7 @@ void ndpi_int_search_thunder_http(struct ndpi_detection_module_struct
 		  "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.0)", 50) == 0) {
       NDPI_LOG(NDPI_PROTOCOL_THUNDER, ndpi_struct, NDPI_LOG_DEBUG,
 	       "Thunder HTTP download detected, adding flow.\n");
-      ndpi_int_thunder_add_connection(ndpi_struct, flow/* , NDPI_CORRELATED_PROTOCOL */);
+      ndpi_int_thunder_add_connection(ndpi_struct, flow);
     }
   }
 }