--- a/src/NetworkInterface.cpp
+++ b/src/NetworkInterface.cpp
@@ -1604,7 +1604,8 @@
 	  ndpi_protocol icmp_proto = flow->get_detected_protocol();
 
 	  if(icmp_proto.category == NDPI_PROTOCOL_CATEGORY_UNSPECIFIED) {
-	    ndpi_fill_ip_protocol_category(ndpi_struct, (struct ndpi_iphdr *)ip, &icmp_proto);
+	    ndpi_fill_ip_protocol_category(ndpi_struct, ((struct ndpi_iphdr *)ip)->saddr,
+					   ((struct ndpi_iphdr*)ip)->daddr, &icmp_proto);
 	    flow->setDetectedProtocol(icmp_proto, false);
 	  }
 	}