diff --git a/server/main-channel.c b/server/main-channel.c
index c2a7b8bb..d69095dc 100644
--- a/server/main-channel.c
+++ b/server/main-channel.c
@@ -902,7 +902,7 @@ static int main_channel_handle_parsed(RedChannelClient *rcc, uint32_t size, uint
         SpiceMsgcMainAgentTokens *tokens;
 
         tokens = (SpiceMsgcMainAgentTokens *)message;
-        reds_on_main_agent_tokens(mcc, tokens->num_tokens);
+        reds_on_main_agent_tokens(reds, mcc, tokens->num_tokens);
         break;
     }
     case SPICE_MSGC_MAIN_ATTACH_CHANNELS:
diff --git a/server/reds.c b/server/reds.c
index 40811136..800f7b3e 100644
--- a/server/reds.c
+++ b/server/reds.c
@@ -940,7 +940,7 @@ void reds_on_main_agent_start(RedsState *reds, MainChannelClient *mcc, uint32_t
     reds->agent_state.write_filter.discard_all = FALSE;
 }
 
-void reds_on_main_agent_tokens(MainChannelClient *mcc, uint32_t num_tokens)
+void reds_on_main_agent_tokens(RedsState *reds, MainChannelClient *mcc, uint32_t num_tokens)
 {
     if (!reds->vdagent) {
         return;
diff --git a/server/reds.h b/server/reds.h
index d1a726f2..ef34e15f 100644
--- a/server/reds.h
+++ b/server/reds.h
@@ -93,7 +93,7 @@ void reds_update_stat_value(uint32_t value);
 /* callbacks from main channel messages */
 
 void reds_on_main_agent_start(RedsState *reds, MainChannelClient *mcc, uint32_t num_tokens);
-void reds_on_main_agent_tokens(MainChannelClient *mcc, uint32_t num_tokens);
+void reds_on_main_agent_tokens(RedsState *reds, MainChannelClient *mcc, uint32_t num_tokens);
 uint8_t *reds_get_agent_data_buffer(RedsState *reds, MainChannelClient *mcc, size_t size);
 void reds_release_agent_data_buffer(RedsState *reds, uint8_t *buf);
 void reds_on_main_agent_data(RedsState *reds, MainChannelClient *mcc, void *message, size_t size);