X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/89ae2e1ddffda5537295f1b0a81c3057b3dc4205..7f3ada717623084ff09274215eae35dfecb95c56:/src/simix/smx_host.c

diff --git a/src/simix/smx_host.c b/src/simix/smx_host.c
index 8bcbf4705b..6ab435081d 100644
--- a/src/simix/smx_host.c
+++ b/src/simix/smx_host.c
@@ -395,7 +395,11 @@ void SIMIX_post_host_execute(smx_action_t action)
 #ifdef HAVE_TRACING
 void SIMIX_set_category(smx_action_t action, const char *category)
 {
-  surf_workstation_model->set_category(action->execution.surf_exec, category);
+  if (action->type == SIMIX_ACTION_EXECUTE){
+    surf_workstation_model->set_category(action->execution.surf_exec, category);
+  }else if (action->type == SIMIX_ACTION_COMMUNICATE){
+    surf_workstation_model->set_category(action->comm.surf_comm, category);
+  }
 }
 #endif