1 files changed, 5 insertions, 10 deletions
diff --git a/lib/bucky.js b/lib/bucky.js
index dddbae9..2488c31 100644
--- a/lib/bucky.js
+++ b/lib/bucky.js
@@ -23,17 +23,12 @@ var bucky = module.exports = {
   ensureFileCountsForThreads: function (req, res, next){
     db.getFileCounts(res.threads_ids).then(function(counts){
       var lookup = {}
-      counts.forEach(function(c,i){
-        res.threads.at(i).set("file_count", c.count)
+      counts.forEach(function(c){
+        lookup[c.thread] = c
       })
-      next()
-    })
-  },
-  ensureFileSizeForThreads: function (req, res, next){
-    db.getFileSizes(res.threads_ids).then(function(sizes){
-      var lookup = {}
-      sizes.forEach(function(c,i){
-        res.threads.at(i).set("file_size", c.size)
+      res.threads.forEach(function(t){
+        var c = lookup[t.id]
+        t.set("file_count", c ? c.count : 0)
       })
       next()
     })