diff --git a/swap/src/protocol/alice.rs b/swap/src/protocol/alice.rs
index 300475a0..6ce5fd95 100644
--- a/swap/src/protocol/alice.rs
+++ b/swap/src/protocol/alice.rs
@@ -26,7 +26,7 @@ pub use self::{
     message1::Message1,
     message2::Message2,
     state::*,
-    swap::swap,
+    swap::{run_until, swap},
 };
 
 mod amounts;
diff --git a/swap/src/protocol/bob.rs b/swap/src/protocol/bob.rs
index 350a0018..ab9bbfc6 100644
--- a/swap/src/protocol/bob.rs
+++ b/swap/src/protocol/bob.rs
@@ -26,7 +26,7 @@ pub use self::{
     message2::Message2,
     message3::Message3,
     state::*,
-    swap::swap,
+    swap::{run_until, swap},
 };
 
 mod amounts;