feat(session): refactor session state

session state is closer to the protocol

1 files changed, 2 insertions, 2 deletions

diff --git a/test/hello.ml b/test/hello.ml
index f4661c3..c7bf46e 100644
--- a/test/hello.ml
+++ b/test/hello.ml
@@ -18,8 +18,8 @@ let main =
   let waiter, wakener = Lwt.wait () in
   try%lwt let* state, connect = create config.sasl.jid.domainpart config in
              S.map (function
-                 | Connected (portal, _) -> portal.push None;
-                                            Lwt.wakeup wakener ()
+                 | Connected (portal, Logged_in _) -> portal.push None;
+                                                      Lwt.wakeup wakener ()
                  | _ -> ())
                state |> S.keep;
              connect ();