diff --git a/services/appflowy-indexer/src/collab_handle.rs b/services/appflowy-indexer/src/collab_handle.rs
index 28ac33d6..daa137e5 100644
--- a/services/appflowy-indexer/src/collab_handle.rs
+++ b/services/appflowy-indexer/src/collab_handle.rs
@@ -287,7 +287,7 @@ mod test {
   use collab::preclude::Collab;
   use collab_document::document::Document;
   use collab_entity::CollabType;
-  use sqlx::Row;
+  use sqlx::{Postgres, Row};
 
   use workspace_template::document::get_started::get_started_document_data;
 
@@ -363,10 +363,10 @@ mod test {
 
     assert_eq!(contents.len(), 1);
 
-    let tokens = sqlx::query_scalar!(
+    let tokens: i64 = sqlx::query_scalar::<Postgres, Option<i64>>(
       "SELECT index_tokens_consumed from af_workspace_ai_usage WHERE workspace_id = $1",
-      workspace_id
     )
+    .bind(workspace_id)
     .fetch_one(&db)
     .await
     .unwrap()