GEDF, PEDF, and PFP: fixed budget hooks

1 files changed, 2 insertions, 3 deletions

diff --git a/litmus/sched_cedf.c b/litmus/sched_cedf.c
index e8de5dd18b7f..1bfdef17e519 100644
--- a/litmus/sched_cedf.c
+++ b/litmus/sched_cedf.c
@@ -860,7 +860,7 @@ static void cedf_tick(struct task_struct* t)
         if (is_realtime(t) &&
                 tsk_rt(t)->budget.ops && budget_quantum_tracked(t) &&
                 budget_exhausted(t)) {
-                        TRACE_TASK(t, "budget exhausted\n");
+                TRACE_TASK(t, "budget exhausted\n");
                 budget_state_machine(t,on_exhausted);
         }
 }
@@ -1499,8 +1499,7 @@ static void cedf_task_exit(struct task_struct * t)
 
         /* disable budget enforcement */
         cedf_untrack_in_top_m(t);
-        if (tsk_rt(t)->budget.ops)
-                budget_state_machine(t,on_exit);
+        budget_state_machine(t,on_exit);
 
 #ifdef CONFIG_REALTIME_AUX_TASKS
         /* make sure we clean up on our way out */