1 files changed, 9 insertions, 2 deletions

diff --git a/litmus/rt_domain.c b/litmus/rt_domain.c
index e6ccf04cecf0..b615092ce9c9 100644
--- a/litmus/rt_domain.c
+++ b/litmus/rt_domain.c
@@ -52,7 +52,11 @@ static unsigned int time2slot(lt_t time)
 static void do_release(struct release_heap *rh)
 {
         unsigned long flags;
-        TS_RELEASE_START;
+
+        if (CRIT_LEVEL_B == rh->dom->level)
+                TS_LVLB_RELEASE_START;
+        else
+                TS_LVLC_RELEASE_START;
 
         raw_spin_lock_irqsave(&rh->dom->release_lock, flags);
         VTRACE("CB has the release_lock 0x%p\n", &rh->dom->release_lock);
@@ -64,7 +68,10 @@ static void do_release(struct release_heap *rh)
         /* call release callback */
         rh->dom->release_jobs(rh->dom, &rh->heap);
 
-        TS_RELEASE_END;
+        if (CRIT_LEVEL_B == rh->dom->level)
+                TS_LVLB_RELEASE_END;
+        else
+                TS_LVLC_RELEASE_END;
 }
 
 #ifdef CONFIG_MERGE_TIMERS