diff --git a/src/lib.rs b/src/lib.rs
index 082d77bfa9d40d481101efa02e087b80434f5322..8674f5eedb3dcf874836957d45bde0a956e0503a 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -9,5 +9,10 @@ pub use common::{ConnectorId, EndpointPolarity, Payload, Polarity, PortId};
 pub use protocol::{ProtocolDescription, TRIVIAL_PD};
 pub use runtime::{error, Connector, DummyLogger, FileLogger, VecLogger};
 
+// TODO: Remove when not benchmarking
+pub use protocol::inputsource::InputSource;
+pub use protocol::ast::Heap;
+pub use protocol::lexer::Lexer;
+
 #[cfg(feature = "ffi")]
 pub mod ffi;