1 files changed, 2 insertions, 15 deletions
diff --git a/lib/generators/site.js b/lib/generators/site.js
index 9b7dbcb..6db2fa4 100644
--- a/lib/generators/site.js
+++ b/lib/generators/site.js
@@ -73,7 +73,7 @@ Generator.prototype.prepareTemplates = function() {
 	);
 
     // Add filter
-    this.env.addFilter("contentLink", this.contentLink.bind(this));
+    this.env.addFilter("contentLink", this.book.contentLink.bind(this.book));
     this.env.addFilter('lvl', function(lvl) {
         return lvl.split(".").length;
     });
@@ -93,24 +93,11 @@ Generator.prototype.finish = function() {
     .then(this.writeSearchIndex);
 };
 
-// Normalize a link to .html and convert README -> index
-Generator.prototype.contentLink = function(link) {
-    if (
-        path.basename(link) == "README"
-        || link == this.book.readmeFile
-    ) {
-        link = path.join(path.dirname(link), "index"+path.extname(link));
-    }
-
-    link = links.changeExtension(link, ".html");
-    return link;
-}
-
 // Convert an input file
 Generator.prototype.writeParsedFile = function(page) {
     var that = this;
 
-    var output = this.contentLink(page.path);
+    var output = this.book.contentLink(page.path);
     output = path.join(that.options.output, output);
 
     var basePath = path.relative(path.dirname(output), this.options.output) || ".";