diff --git a/src/runtime2/component/mod.rs b/src/runtime2/component/mod.rs
index d5f0016f42f5f6714cb03641d63a8d7f1140acbb..71745de1d588b5c2479bd851d1c6e6ef99f05f24 100644
--- a/src/runtime2/component/mod.rs
+++ b/src/runtime2/component/mod.rs
@@ -16,7 +16,7 @@ use super::runtime::*;
 /// If the component is sleeping, then that flag will be atomically set to
 /// false. If we're the ones that made that happen then we add it to the work
 /// queue.
-pub(crate) fn wake_up_if_sleeping(sched_ctx: &SchedulerCtx, comp_id: CompId, handle: &CompHandle) {
+pub(crate) fn wake_up_if_sleeping(runtime: &RuntimeInner, comp_id: CompId, handle: &CompHandle) {
     use std::sync::atomic::Ordering;
 
     let should_wake_up = handle.sleeping
@@ -25,6 +25,6 @@ pub(crate) fn wake_up_if_sleeping(sched_ctx: &SchedulerCtx, comp_id: CompId, han
 
     if should_wake_up {
         let comp_key = unsafe{ comp_id.upgrade() };
-        sched_ctx.runtime.enqueue_work(comp_key);
+        runtime.enqueue_work(comp_key);
     }
 }
\ No newline at end of file