diff --git a/src/runtime2/tests/mod.rs b/src/runtime2/tests/mod.rs
index a5c2aac79bf56dedc089cee2118064d8f49f87c7..c3ac4c825bd954e54e1c2ab043465a0fb8f064a5 100644
--- a/src/runtime2/tests/mod.rs
+++ b/src/runtime2/tests/mod.rs
@@ -1,3 +1,5 @@
+mod network_shapes;
+
 use super::*;
 use crate::{PortId, ProtocolDescription};
 use crate::common::Id;
@@ -75,7 +77,7 @@ fn test_put_and_get() {
 
     let thing = TestTimer::new("put_and_get");
     run_test_in_runtime(CODE, |api| {
-        let channel = api.create_channel();
+        let channel = api.create_channel().unwrap();
 
         api.create_connector("", "putter", ValueGroup::new_stack(vec![
             Value::Output(PortId(Id{ connector_id: 0, u32_suffix: channel.putter_id.index })),
@@ -150,7 +152,7 @@ fn test_star_shaped_request() {
         api.create_connector("", "constructor", ValueGroup::new_stack(vec![
             Value::UInt32(5),
             Value::UInt32(NUM_LOOPS),
-        ]));
+        ])).expect("create connector");
     });
 }
 
@@ -227,6 +229,6 @@ fn test_conga_line_request() {
         api.create_connector("", "constructor", ValueGroup::new_stack(vec![
             Value::UInt32(5),
             Value::UInt32(NUM_LOOPS)
-        ]));
+        ])).expect("create connector");
     });
 }
\ No newline at end of file