diff --git a/src/asfvolt16_driver.h b/src/asfvolt16_driver.h
index 6f5a884..6871e2b 100755
--- a/src/asfvolt16_driver.h
+++ b/src/asfvolt16_driver.h
@@ -71,7 +71,8 @@
 extern uint32_t asfvolt16_bal_cfg_clear(BalKey *key);
 extern uint32_t asfvolt16_bal_cfg_get(BalKey *key, BalCfg *cfg);
 extern void asfvolt16_send_omci_msg(uint16_t olt_no, uint32_t onu_nu, char* packet_to_send);
-extern uint32_t asfvolt16_bal_stats_get(BalIntfType intf_type, uint32_t intf_id, BalInterfaceStatData *statData);
+extern uint32_t asfvolt16_bal_stats_get(BalIntfType intf_type, uint32_t intf_id, BalInterfaceStatData *statData,
+                                        BalInterfaceKey *statKey);
 grpc_c_client_t *client;
 
 #endif /* DRIVERAPP_ASFVOLT16_DRIVER */