1 files changed, 7 insertions, 6 deletions
diff --git a/src/wireguard/mod.rs b/src/wireguard/mod.rs
index 9417e57..563a22f 100644
--- a/src/wireguard/mod.rs
+++ b/src/wireguard/mod.rs
@@ -1,7 +1,6 @@
-mod wireguard;
-// mod config;
 mod constants;
 mod timers;
+mod wireguard;
 
 mod handshake;
 mod router;
@@ -12,12 +11,14 @@ mod tests;
 
 /// The WireGuard sub-module contains a pure, configurable implementation of WireGuard.
 /// The implementation is generic over:
-/// 
+///
 /// - TUN type, specifying how packets are received on the interface side: a reader/writer and MTU reporting interface.
 /// - Bind type, specifying how WireGuard messages are sent/received from the internet and what constitutes an "endpoint"
-
-pub use wireguard::{Wireguard, Peer};
+pub use wireguard::{Peer, Wireguard};
 
 pub use types::bind;
 pub use types::tun;
-pub use types::Endpoint;
-\ No newline at end of file
+pub use types::Endpoint;
+
+#[cfg(test)]
+pub use types::dummy;