X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/983883d65cbcf86c675a323208f9dd15ffbbf583..67e5c5c9a1dddb2528b23194e6001bc8b70ce582:/src/mc/mc_client_api.c

diff --git a/src/mc/mc_client_api.c b/src/mc/mc_client_api.c
index 9c7b2e7619..fad7c0ffd8 100644
--- a/src/mc/mc_client_api.c
+++ b/src/mc/mc_client_api.c
@@ -68,9 +68,7 @@ void MC_ignore(void* addr, size_t size)
   }
 
   // TODO, remove this once the migration has been completed
-  int raw_mem_set = (mmalloc_get_current_heap() == mc_heap);
-  MC_SET_MC_HEAP;
+  xbt_mheap_t heap = mmalloc_set_current_heap(mc_heap);
   MC_process_ignore_memory(&mc_model_checker->process, addr, size);
-  if (!raw_mem_set)
-    MC_SET_STD_HEAP;
+  mmalloc_set_current_heap(heap);
 }