1 files changed, 2 insertions, 0 deletions

diff --git a/apps/plugins/battery_bench.c b/apps/plugins/battery_bench.c
index 13ce172f67..fef3f36db2 100644
--- a/apps/plugins/battery_bench.c
+++ b/apps/plugins/battery_bench.c
@@ -20,6 +20,7 @@
 #ifndef SIMULATOR /* not for the simulator */
 
 #include "plugin.h"
+#include "mem_function_wrappers.h"
 PLUGIN_HEADER
 
 #define BATTERY_LOG "/battery_bench.txt"
@@ -98,6 +99,7 @@ PLUGIN_HEADER
 
 /****************************** Plugin Entry Point ****************************/
 static struct plugin_api* rb;
+MEM_FUNCTION_WRAPPERS(rb);
 int main(void);
 bool exit_tsr(bool);
 void thread(void);