diff --git a/src/lib.rs b/src/lib.rs
index 891db6c02e70a9780f297b70f089fa6438177a3d..357f234aa26047196d17a6c8e5f2a3727e53f6b3 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,19 +1,9 @@
 #[macro_use]
 mod macros;
 
-mod common;
+// mod common;
 mod protocol;
-mod runtime;
-pub mod runtime2;
+pub mod runtime;
 mod collections;
 
-pub use common::{ConnectorId, EndpointPolarity, Payload, Polarity, PortId};
-pub use protocol::ProtocolDescription;
-pub use runtime::{error, Connector, DummyLogger, FileLogger, VecLogger};
-
-// TODO: Remove when not benchmarking
-pub use protocol::input_source::InputSource;
-pub use protocol::ast::Heap;
-
-#[cfg(feature = "ffi")]
-pub mod ffi;
+pub use protocol::ProtocolDescription;
\ No newline at end of file