Fixed edge case

1 files changed, 2 insertions, 1 deletions

diff --git a/litmus/reservations/gedf_reservation.c b/litmus/reservations/gedf_reservation.c
index 9f02617821a1..1649eaf026d7 100644
--- a/litmus/reservations/gedf_reservation.c
+++ b/litmus/reservations/gedf_reservation.c
@@ -249,7 +249,7 @@ static void gedf_task_replenish_budget(
 {
         struct task_struct* t = ((struct gedf_task_reservation*)res)->task;
 
-        if (is_completed(t)) {
+        if (is_completed(t) || (budget_enforced(t) && budget_exhausted(t))) {
                 sched_trace_task_completion(t, 0);
                 prepare_for_next_period(t);
                 tsk_rt(t)->completed = 0;
@@ -271,6 +271,7 @@ static void gedf_drain_budget(
         lt_t how_much,
         int cpu)
 {
+
         if (how_much > res->cur_budget)
                 res->cur_budget = 0;
         else