diff --git a/src/tbf_dl.h b/src/tbf_dl.h
index 3a7c41d..8d4d716 100644
--- a/src/tbf_dl.h
+++ b/src/tbf_dl.h
@@ -21,6 +21,14 @@
 
 #include <stdint.h>
 
+#ifdef __cplusplus
+extern "C" {
+#endif
+#include <tbf_fsm.h>
+#ifdef __cplusplus
+}
+#endif
+
 /*
  * TBF instance
  */
@@ -78,6 +86,8 @@
 	struct rate_ctr_group *m_dl_gprs_ctrs;
 	struct rate_ctr_group *m_dl_egprs_ctrs;
 
+	struct tbf_dl_fsm_ctx state_fsm;
+
 protected:
 	struct ana_result {
 		unsigned received_packets;