Fix unwrap on missing task

1 files changed, 3 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 7bce3c7..b11ee27 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -239,7 +239,9 @@ impl<'a> Scheduler<'a> {
             self.dispatch_next_task();
 
             for (pid, parent) in self.reciever.try_iter() {
-                self.managed_tasks.get_mut(&(pid as u32)).unwrap().parent = parent as u32;
+                if let Some(task) = self.managed_tasks.get_mut(&(pid as u32)) {
+                    task.parent = parent as u32;
+                }
             }
 
             // If no task is ready to run (or in case of error), stop dispatching tasks and notify