Instead calling diff_collection.count use diff_collection.size which is cache on the diff_collection