1 files changed, 2 insertions, 5 deletions

diff --git a/litmus/reservations/gedf_reservation.c b/litmus/reservations/gedf_reservation.c
index 65ca2b8c9fd6..a08373e28469 100644
--- a/litmus/reservations/gedf_reservation.c
+++ b/litmus/reservations/gedf_reservation.c
@@ -226,10 +226,7 @@ static struct task_struct* gedf_task_dispatch_client(
         int cpu)
 {
         struct gedf_task_reservation* tmp = (struct gedf_task_reservation*)res;
-        if (budget_exhausted(tmp->task))
-                return NULL;
-        else
-                return tmp->task;
+        return tmp->task;
 }
 
 static void gedf_replenish_budget(
@@ -250,7 +247,7 @@ static void gedf_task_replenish_budget(
 {
         struct task_struct* t = ((struct gedf_task_reservation*)res)->task;
 
-        if (is_completed(t) || (budget_enforced(t) && budget_exhausted(t))) {
+        if (is_completed(t)) {
                 sched_trace_task_completion(t, 0);
                 prepare_for_next_period(t);
                 tsk_rt(t)->completed = 0;