diff --git a/src/tests/mod.rs b/src/tests/mod.rs
index 3a21d0b..0ff4e1b 100644
--- a/src/tests/mod.rs
+++ b/src/tests/mod.rs
@@ -3,4 +3,6 @@
 pub mod tests;
 
 #[cfg(test)]
-pub mod stream_states_tests;
\ No newline at end of file
+pub mod stream_states_tests;
+#[cfg(test)]
+pub mod message_handler_tests;
\ No newline at end of file