diff --git a/test/client_test/session_test.gleam b/test/client_test/session_test.gleam
index 74b563c..e0e7dc3 100644
--- a/test/client_test/session_test.gleam
+++ b/test/client_test/session_test.gleam
@@ -2,25 +2,26 @@ import client
 import client/route
 import client/session
 import client/user/role
+import client_test as testing
 import client_test/dummy
+import lustre/dev/simulate
 import rsvp
 
 pub fn session_restored_success_test() -> Nil {
-  let on_success = route.Dashboard
-  let on_failure = route.Login
-  let pending = session.Pending(on_success:, on_failure:)
-  let route = route.Home
+  let simulation =
+    route.Home
+    |> testing.init
+    |> testing.simulation
 
-  let #(model, _) = dummy.model(session: pending, route:)
-  let restored = dummy.session(role.Default)
-  let msg = client.UserRestoredSession(Ok(restored))
+  let session = dummy.session(role.Default)
+  let message = client.UserRestoredSession(Ok(session))
 
-  let #(model, _) = client.update(model, msg)
-
-  assert model.session == restored as "session restored"
-  assert model.route == on_success as "user redirected"
+  let model =
+    simulation
+    |> simulate.message(message)
+    |> simulate.model
 
-  Nil
+  assert model.session == session as "restored"
 }
 
 pub fn session_restored_failure_test() -> Nil {