1 files changed, 9 insertions, 2 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 301390148d02..40340dfa9d67 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -302,7 +302,12 @@ static void reinit_litmus_state(struct task_struct* p, int restore)
         /* We probably should not be inheriting any task's priority
          * at this point in time.
          */
-        WARN_ON(p->rt_param.inh_task);
+        WARN_ON(p->rt_param.eff_prio);
+    
+#ifdef CONFIG_LITMUS_NESTED_LOCKING
+    WARN_ON(p->rt_param.local_prio);
+    WARN_ON(p->rt_param.trans_prio);
+#endif
 
         /* Cleanup everything else. */
         memset(&p->rt_param, 0, sizeof(p->rt_param));
@@ -468,7 +473,9 @@ void litmus_exec(void)
         struct task_struct* p = current;
 
         if (is_realtime(p)) {
-                WARN_ON(p->rt_param.inh_task);
+                WARN_ON(p->rt_param.eff_prio);
+        WARN_ON(p->rt_param.local_prio);
+        WARN_ON(p->rt_param.trans_prio);
                 if (tsk_rt(p)->ctrl_page) {
                         free_page((unsigned long) tsk_rt(p)->ctrl_page);
                         tsk_rt(p)->ctrl_page = NULL;