1 files changed, 3 insertions, 6 deletions
diff --git a/src/include/ndpi_main.h b/src/include/ndpi_main.h
index d262a5108..7012fd8ec 100644
--- a/src/include/ndpi_main.h
+++ b/src/include/ndpi_main.h
@@ -106,13 +106,10 @@ extern "C" {
 			       ndpi_protocol_category_t protoCategory,
 			       ndpi_port_range *tcpDefPorts,
 			       ndpi_port_range *udpDefPorts);
-  void ndpi_set_risk(struct ndpi_detection_module_struct *ndpi_str,
-		     struct ndpi_flow_struct *flow, ndpi_risk_enum r,
+  void ndpi_set_risk(struct ndpi_flow_struct *flow, ndpi_risk_enum r,
 		     char *risk_message);
-  void ndpi_unset_risk(struct ndpi_detection_module_struct *ndpi_str,
-		       struct ndpi_flow_struct *flow, ndpi_risk_enum r);    
-  int ndpi_isset_risk(struct ndpi_detection_module_struct *ndpi_str,
-		      struct ndpi_flow_struct *flow, ndpi_risk_enum r);
+  void ndpi_unset_risk(struct ndpi_flow_struct *flow, ndpi_risk_enum r);
+  int ndpi_isset_risk(struct ndpi_flow_struct *flow, ndpi_risk_enum r);
   int ndpi_is_printable_buffer(u_int8_t const * const buf, size_t len);
   int ndpi_normalize_printable_string(char * const str, size_t len);
   int ndpi_is_valid_hostname(char * const str, size_t len);