diff --git a/include/fluent-bit/flb_task.h b/include/fluent-bit/flb_task.h
index 85bbe6f1280..719d245ec44 100644
--- a/include/fluent-bit/flb_task.h
+++ b/include/fluent-bit/flb_task.h
@@ -125,6 +125,7 @@ struct flb_task_enqueued {
 
 int flb_task_running_count(struct flb_config *config);
 int flb_task_running_print(struct flb_config *config);
+int flb_task_map_get_task_id(struct flb_config *config);
 
 struct flb_task *flb_task_create(uint64_t ref_id,
                                  const char *buf,
diff --git a/src/flb_task.c b/src/flb_task.c
index 53bb7dffd92..b37196375fe 100644
--- a/src/flb_task.c
+++ b/src/flb_task.c
@@ -331,6 +331,10 @@ int flb_task_running_print(struct flb_config *config)
     return 0;
 }
 
+int flb_task_map_get_task_id(struct flb_config *config) {
+    return map_get_task_id(config);
+}
+
 /* Create an engine task to handle the output plugin flushing work */
 struct flb_task *flb_task_create(uint64_t ref_id,
                                  const char *buf,