diff --git a/src/tests/mod.rs b/src/tests/mod.rs
index b33b440..6051632 100644
--- a/src/tests/mod.rs
+++ b/src/tests/mod.rs
@@ -9,4 +9,6 @@ pub mod message_handler_tests;
 #[cfg(test)]
 pub mod states_io_handler_tests;
 #[cfg(test)]
-pub mod socket_handler_tests;
\ No newline at end of file
+pub mod socket_handler_tests;
+#[cfg(test)]
+pub mod state_update_tests;
\ No newline at end of file
diff --git a/src/tests/state_update_tests.rs b/src/tests/state_update_tests.rs
new file mode 100644
index 0000000..e69de29