1 files changed, 6 insertions, 2 deletions
diff --git a/filamento/src/logic/connect.rs b/filamento/src/logic/connect.rs
index dc05448..9d61ca4 100644
--- a/filamento/src/logic/connect.rs
+++ b/filamento/src/logic/connect.rs
@@ -5,17 +5,21 @@ use tracing::debug;
 use crate::{
     Command, UpdateMessage,
     error::{ConnectionJobError, Error, RosterError},
+    files::FileStore,
     presence::{Online, PresenceType},
 };
 
 use super::ClientLogic;
 
-pub async fn handle_connect(logic: ClientLogic, connection: Connected) {
+pub async fn handle_connect<Fs: FileStore + Clone + Send + Sync>(
+    logic: ClientLogic<Fs>,
+    connection: Connected,
+) {
     let (send, recv) = oneshot::channel();
     debug!("getting roster");
     logic
         .clone()
-        .handle_online(Command::GetRoster(send), connection.clone())
+        .handle_online(Command::GetRosterWithUsers(send), connection.clone())
         .await;
     debug!("sent roster req");
     let roster = recv.await;