1 files changed, 5 insertions, 2 deletions
diff --git a/iosApp/iosApp/Session/RootViewModel.swift b/iosApp/iosApp/Session/RootViewModel.swift
index 682deee..ec103ba 100644
--- a/iosApp/iosApp/Session/RootViewModel.swift
+++ b/iosApp/iosApp/Session/RootViewModel.swift
@@ -18,6 +18,7 @@
 import Foundation
 import shared
 
+@MainActor
 class RootViewModel: ObservableObject {
     @Inject private var sessionController: SessionController
     
@@ -25,13 +26,15 @@ class RootViewModel: ObservableObject {
     
     init() {
         let collector = Collector<SessionController.LoginState?>(callback: setLoginState)
-        sessionController.loginStateFlow.collect(collector: collector) { (unit, error) in }
+        sessionController.loginStateFlow.collect(collector: collector) { _ in }
         restoreSession()
     }
     
     func setLoginState(state: SessionController.LoginState?) {
         print("State is: \(state?.debugDescription ?? "")")
-        self.loginState = state ?? SessionController.LoginStateNothing()
+        Task { @MainActor in
+            self.loginState = state ?? SessionController.LoginStateNothing()
+        }
     }
     
     func restoreSession() {