1 files changed, 2 insertions, 0 deletions

diff --git a/litmus/sched_litmus.c b/litmus/sched_litmus.c
index 52afec278ec8..3ed713b8791b 100644
--- a/litmus/sched_litmus.c
+++ b/litmus/sched_litmus.c
@@ -1,6 +1,7 @@
 /* This file is included from kernel/sched.c */
 
 #include <litmus/litmus.h>
+#include <litmus/budget.h>
 #include <litmus/sched_plugin.h>
 
 static void update_time_litmus(struct rq *rq, struct task_struct *p)
@@ -148,6 +149,7 @@ litmus_schedule(struct rq *rq, struct task_struct *prev)
                 next->se.exec_start = rq->clock;
         }
 
+        update_enforcement_timer(next);
         return next;
 }