1 files changed, 1 insertions, 1 deletions
diff --git a/include/query/irs.h b/include/query/irs.h
index e2d9325..51eb4e2 100644
--- a/include/query/irs.h
+++ b/include/query/irs.h
@@ -103,7 +103,7 @@ public:
             weights.push_back((bs) ? bs->records.size() : 0);
         }
 
-        size_t total_weight = 0;
+        size_t total_weight = weights[0];
         for (auto &s : shard_states) {
             auto state = (State<R> *) s;
             total_weight += state->total_weight;