diff --git a/src/runtime2/mod.rs b/src/runtime2/mod.rs
index 93d9db7f74210151b07b2858bb925c97c1ede825..6ec88c480c45afdf81dd51a0bf1bc7810fc2bc4e 100644
--- a/src/runtime2/mod.rs
+++ b/src/runtime2/mod.rs
@@ -7,4 +7,4 @@ mod scheduler;
 
 pub use runtime::Runtime;
 pub(crate) use scheduler::SchedulerCtx;
-pub(crate) use communication::Message;
\ No newline at end of file
+pub(crate) use communication::{Message, ControlMessage, SyncMessage, DataMessage};
\ No newline at end of file