diff --git a/src/runtime2/scheduler.rs b/src/runtime2/scheduler.rs
index 047cd25f8717f6f22e55404210a125a5d2ad56ef..bf620950f3942fc085ebf3d4db60f089afd0f2ae 100644
--- a/src/runtime2/scheduler.rs
+++ b/src/runtime2/scheduler.rs
@@ -2,6 +2,7 @@ use std::sync::atomic::Ordering;
 
 use super::component::*;
 use super::runtime::*;
+use super::communication::*;
 
 /// Data associated with a scheduler thread
 pub(crate) struct Scheduler {
@@ -38,7 +39,7 @@ impl Scheduler {
             // be re-executed immediately.
             let mut new_scheduling = CompScheduling::Immediate;
             while let CompScheduling::Immediate = new_scheduling {
-                new_scheduling = component.private.code.run(&scheduler_ctx, &mut component.private.ctx).expect("TODO: Handle error");
+                new_scheduling = component.code.run(&scheduler_ctx, &mut component.private.ctx).expect("TODO: Handle error");
             }
 
             // Handle the new scheduling