diff --git a/msmarco-passage-ranking/track.py b/msmarco-passage-ranking/track.py
index 71f999df2..dcf5055d1 100644
--- a/msmarco-passage-ranking/track.py
+++ b/msmarco-passage-ranking/track.py
@@ -165,11 +165,11 @@ def params(self):
             raise Exception(f"The query strategy \\`{self._query_strategy}]\\` is not implemented")
 
         self._iters = (self._iters + 1) % len(self._queries)
+        query["track_total_hits"] = self._track_total_hits
+        query["size"] = self._size
         return {
             "index": self._index_name,
             "cache": self._cache,
-            "size": self._size,
-            "track_total_hits": self._track_total_hits,
             "body": query,
         }
 
diff --git a/search/mteb/dbpedia/track.py b/search/mteb/dbpedia/track.py
index b5c4ae24a..4405d8f47 100644
--- a/search/mteb/dbpedia/track.py
+++ b/search/mteb/dbpedia/track.py
@@ -103,13 +103,13 @@ def partition(self, partition_index, total_partitions):
     def params(self):
         query_obj = self._queries[self._iters]
         query = generate_query(query_obj["text"], self._title_field, self._title_boost, self._text_field, self._text_boost)
+        query["track_total_hits"] = self._track_total_hits
+        query["size"] = self._size
 
         self._iters = (self._iters + 1) % len(self._queries)
         return {
             "index": self._index_name,
             "cache": self._cache,
-            "size": self._size,
-            "track_total_hits": self._track_total_hits,
             "body": query,
         }