1 files changed, 6 insertions, 10 deletions
diff --git a/lib/models/book.js b/lib/models/book.js
index 0a4a41a..2a97f78 100644
--- a/lib/models/book.js
+++ b/lib/models/book.js
@@ -176,18 +176,14 @@ Book.prototype.isLanguageBook = function() {
 };
 
 /**
-    Returns the book for a language.
-    Returns itself if matches the language itself.
+    Return a languages book
 
-    @param {String} lang Language ID
-    @return {Book | Null} Null if no such language
+    @param {String} language
+    @return {Book}
 */
-Book.prototype.getLanguageBook = function(lang) {
-    if (this.getLanguage() === lang) {
-        return this;
-    } else {
-        return this.getBooks().get(lang) || null;
-    }
+Book.prototype.getLanguageBook = function(language) {
+    var books = this.getBooks();
+    return books.get(language);
 };
 
 /**