[SQUASH ME] Make it work.

kernel/exit.c

@@ -772,6 +772,7 @@ void __noreturn do_exit(long code)
 
 	rq = this_rq();
 
+	//FIXME: This disclaimer is not really true anymore (since i dont follow it and it works either way)
 	/**
 	 * BEWARE:
 	 * We want to update the plan runtime model not too early to ensure that the exiting process

kernel/sched/pb.c

@@ -291,10 +291,10 @@ static void dequeue_task_pb(struct rq *rq, struct task_struct *p, int flags)
 
 	bool process_exited = false;
 
-	printk("Dequeue task: %u\n", p->pid);
+	//printk("Dequeue task: %u\n", p->pid);
 
-	if (pb->waiting_on_io) {
-		printk("Waiting for IO\n", p->pid);
+	if (pb->waiting_on_io && p->state != TASK_DEAD) {
+		printk("Waiting for IO\n");
 		return;
 	}
 	pb->waiting_on_io = 1;