diff --git a/runtime/starpu/codelets/codelet_ztsmqr.c b/runtime/starpu/codelets/codelet_ztsmqr.c
index 55468843244327b50d7d2a307d17a5e634d4a95d..282e58d72a7ad0851da82d45ce362d3d3b049a4a 100644
--- a/runtime/starpu/codelets/codelet_ztsmqr.c
+++ b/runtime/starpu/codelets/codelet_ztsmqr.c
@@ -173,6 +173,7 @@ void MORSE_TASK_ztsmqr(MORSE_option_t *options,
             STARPU_VALUE,    &ldwork,            sizeof(int),
             STARPU_PRIORITY,  options->priority,
             STARPU_CALLBACK,  callback,
+            STARPU_EXECUTE_ON_NODE, A2->get_rankof(A2, A2m, A2n),
             0);
     }
 }
diff --git a/runtime/starpu/codelets/codelet_ztsqrt.c b/runtime/starpu/codelets/codelet_ztsqrt.c
index 84b37b64bd63ab4093274154ec2f316a27a49d5d..240605594b3124a56da6c4dd392cb90f66f1215c 100644
--- a/runtime/starpu/codelets/codelet_ztsqrt.c
+++ b/runtime/starpu/codelets/codelet_ztsqrt.c
@@ -129,6 +129,7 @@ void MORSE_TASK_ztsqrt(MORSE_option_t *options,
             STARPU_VALUE,    &h_work,            sizeof(MORSE_starpu_ws_t *),
             STARPU_PRIORITY,  options->priority,
             STARPU_CALLBACK,  callback,
+            STARPU_EXECUTE_ON_NODE, A2->get_rankof(A2, A2m, A2n),
             0);
     }
 }