1 files changed, 5 insertions, 2 deletions
diff --git a/filamento/src/logic/disconnect.rs b/filamento/src/logic/disconnect.rs
index 241c3e6..ebcfd4f 100644
--- a/filamento/src/logic/disconnect.rs
+++ b/filamento/src/logic/disconnect.rs
@@ -1,11 +1,14 @@
 use lampada::Connected;
 use stanza::client::Stanza;
 
-use crate::{UpdateMessage, presence::Offline};
+use crate::{UpdateMessage, files::FileStore, presence::Offline};
 
 use super::ClientLogic;
 
-pub async fn handle_disconnect(logic: ClientLogic, connection: Connected) {
+pub async fn handle_disconnect<Fs: FileStore + Clone>(
+    logic: ClientLogic<Fs>,
+    connection: Connected,
+) {
     // TODO: be able to set offline status message
     let offline_presence: stanza::client::presence::Presence = Offline::default().into_stanza(None);
     let stanza = Stanza::Presence(offline_presence);