diff --git a/runtime/starpu/codelets/codelet_map.c b/runtime/starpu/codelets/codelet_map.c
index f6883c5f0d15ac764724ef7b209accf4b931c7fd..b6b16d19041a19050f343cc9c65ee2be779b2ee5 100644
--- a/runtime/starpu/codelets/codelet_map.c
+++ b/runtime/starpu/codelets/codelet_map.c
@@ -17,6 +17,8 @@
 #include "chameleon_starpu.h"
 #include "runtime_codelet_z.h"
 
+CHAMELEON_CL_CB(map, starpu_matrix_get_nx(task->handles[0]), starpu_matrix_get_ny(task->handles[0]), 0, M*N);
+
 #if !defined(CHAMELEON_SIMULATION)
 static void cl_map_cpu_func(void *descr[], void *cl_arg)
 {