diff --git a/packages/api-server/api_server/repositories/tasks.py b/packages/api-server/api_server/repositories/tasks.py
index 4e83c9770..5fb1ee963 100644
--- a/packages/api-server/api_server/repositories/tasks.py
+++ b/packages/api-server/api_server/repositories/tasks.py
@@ -147,6 +147,7 @@ async def query_task_states(
                 query = add_pagination(
                     query,
                     pagination,
+                    # TODO(koonpeng): remove this mapping after `pickup` and `destination` query is removed.
                     {"pickup": "label=pickup", "destination": "label=destination"},
                     group_by="labels__state_id",
                 )