diff --git a/src/runtime2/scheduler.rs b/src/runtime2/scheduler.rs
index e7359488f13d940ccc259ceabb1cb744c0b6de23..87bdb80e316c95be1016a1347b9f6c5aaaaba5fa 100644
--- a/src/runtime2/scheduler.rs
+++ b/src/runtime2/scheduler.rs
@@ -61,9 +61,9 @@ impl Scheduler {
             let mut new_scheduling = CompScheduling::Immediate;
             while let CompScheduling::Immediate = new_scheduling {
                 while let Some(message) = component.inbox.pop() {
-                    component.code.handle_message(&mut scheduler_ctx, &mut component.ctx, message);
+                    component.component.handle_message(&mut scheduler_ctx, &mut component.ctx, message);
                 }
-                new_scheduling = component.code.run(&mut scheduler_ctx, &mut component.ctx).expect("TODO: Handle error");
+                new_scheduling = component.component.run(&mut scheduler_ctx, &mut component.ctx).expect("TODO: Handle error");
             }
 
             // Handle the new scheduling