1 files changed, 8 insertions, 8 deletions
diff --git a/MatrixUtils.Web/Pages/Tools/Info/SessionCount.razor b/MatrixUtils.Web/Pages/Tools/Info/SessionCount.razor
index 3b68bfa..ce3513b 100644
--- a/MatrixUtils.Web/Pages/Tools/Info/SessionCount.razor
+++ b/MatrixUtils.Web/Pages/Tools/Info/SessionCount.razor
@@ -4,7 +4,7 @@
 @using System.Collections.ObjectModel
 @using LibMatrix
 @using System.Collections.Frozen
-@using LibMatrix.EventTypes.Spec.State
+@using LibMatrix.EventTypes.Spec.State.RoomInfo
 <h3>User Trace</h3>
 <hr/>
 
@@ -73,20 +73,20 @@
 
     protected override async Task OnInitializedAsync() {
         log.CollectionChanged += (sender, args) => StateHasChanged();
-        var hs = await RMUStorage.GetCurrentSessionOrNavigate();
+        var hs = await sessionStore.GetCurrentHomeserver(navigateOnFailure: true);
         if (hs is null) return;
         rooms.CollectionChanged += (sender, args) => StateHasChanged();
-        var sessions = await RMUStorage.GetAllTokens();
+        var sessions = await sessionStore.GetAllSessions();
         foreach (var userAuth in sessions) {
-            var session = await RMUStorage.GetSession(userAuth);
-            if (session is not null) {
-                var sessionRooms = await session.GetJoinedRooms();
+            var homeserver = await sessionStore.GetHomeserver(userAuth.Key);
+            if (homeserver is not null) {
+                var sessionRooms = await homeserver.GetJoinedRooms();
                 foreach (var room in sessionRooms) {
                     rooms.Add(room);
                 }
 
                 StateHasChanged();
-                log.Add($"Got {sessionRooms.Count} rooms for {userAuth.UserId}");
+                log.Add($"Got {sessionRooms.Count} rooms for {userAuth.Value.Auth.UserId}");
             }
         }
 
@@ -106,7 +106,7 @@
 
         log.Add($"Done fetching members!");
 
-        UserIDs.RemoveAll(x => sessions.Any(y => y.UserId == x));
+        UserIDs.RemoveAll(x => sessions.Any(y => y.Value.Auth.UserId == x));
 
         StateHasChanged();
         Console.WriteLine("Rerendered!");