diff --git a/src/tests/main_tests.rs b/src/tests/main_tests.rs
index 88b5cef..7396d03 100644
--- a/src/tests/main_tests.rs
+++ b/src/tests/main_tests.rs
@@ -1,5 +1,7 @@
 use workctl::sync_flag;
 
+use crate::modules::{message_handler::MessageHandler, stream_states::{ stream_states_class::StreamState, enums::Scenes, state_update::StateUpdate}};
+
 
 #[test]
 fn it_works() {
@@ -9,8 +11,17 @@ fn it_works() {
 
 #[test]
 fn can_make_ctrl_c_handler() {
-    let (control_c_flag_tx, control_c_called_flag_rx) = sync_flag::new_syncflag(false);
+    let (control_c_flag_tx, _control_c_called_flag_rx) = sync_flag::new_syncflag(false);
     crate::setup_control_c(control_c_flag_tx);
+    drop(_control_c_called_flag_rx);
 }
 
+#[test]
+fn test_updating_state_from_state_update() {
+    let mut state = StreamState::new();
+    let update = StateUpdate::Scene(Scenes::Augmented);
 
+    //Update handled extensivly in stream_state_tests
+    state.handle_update(update);
+    assert_eq!(state.current_scene, Scenes::Augmented);
+}