diff --git a/test_utilities/src/lib.rs b/test_utilities/src/lib.rs
index 0fee9ef..2e3d49b 100644
--- a/test_utilities/src/lib.rs
+++ b/test_utilities/src/lib.rs
@@ -124,7 +124,7 @@ pub mod database {
 }
 
 pub mod notification {
-    use application::request::models::Current;
+    use application::request::models::{Current, State};
     use application::request::notification::Notification;
 
     pub struct FakeNotification<T>
@@ -138,7 +138,7 @@ pub mod notification {
     where
         T: Fn(),
     {
-        fn send(&self, currents: &Vec<Current>) {
+        fn send(&self, currents: &Vec<State>) {
             (self.callback)();
         }
     }