Add duration to final log for generation

1 files changed, 3 insertions, 5 deletions

diff --git a/lib/models/summaryArticle.js b/lib/models/summaryArticle.js
index d6cd045..ec1104c 100644
--- a/lib/models/summaryArticle.js
+++ b/lib/models/summaryArticle.js
@@ -11,7 +11,7 @@ var SummaryArticle = Immutable.Record({
     title:      String(),
     ref:        String(),
     articles:   Immutable.List()
-});
+}, 'SummaryArticle');
 
 SummaryArticle.prototype.getLevel = function() {
     return this.get('level');
@@ -125,12 +125,10 @@ SummaryArticle.create = function(def, level) {
 
     @param {Article|Part} base
     @param {Function(article)} iter
-    @param {String} method
     @return {Article}
 */
-SummaryArticle.findArticle = function(base, iter, method) {
-    method = method || 'getArticles';
-    var articles = base[method]();
+SummaryArticle.findArticle = function(base, iter) {
+    var articles = base.getArticles();
 
     return articles.reduce(function(result, article) {
         if (result) return result;