4 files changed, 17 insertions, 29 deletions
diff --git a/lib/modifiers/config/addPlugin.js b/lib/modifiers/config/addPlugin.js
index 29f118a..b495062 100644
--- a/lib/modifiers/config/addPlugin.js
+++ b/lib/modifiers/config/addPlugin.js
@@ -2,21 +2,18 @@ var PluginDependency = require('../../models/pluginDependency');
 
 /**
  * Add a plugin to a book's configuration
- * @param {Book} book
+ * @param {Config} config
  * @param {String} plugin
  * @param {String} version (optional)
- * @return {Book}
+ * @return {Config}
  */
-function addPlugin(book, plugin, version) {
-    var config = book.getConfig();
+function addPlugin(config, plugin, version) {
     var deps = config.getPluginDependencies();
 
     var dep = PluginDependency.create(plugin, version);
 
     deps = deps.push(dep);
-    config = config.setPluginDependencies(deps);
-
-    return book.setConfig(config);
+    return config.setPluginDependencies(deps);
 }
 
 module.exports = addPlugin;
diff --git a/lib/modifiers/config/editPlugin.js b/lib/modifiers/config/editPlugin.js
index 71549f2..f9b6551 100644
--- a/lib/modifiers/config/editPlugin.js
+++ b/lib/modifiers/config/editPlugin.js
@@ -1,16 +1,13 @@
 
 /**
- * Edit ocnfiguration of a plugin
- * @param {Book} book
+ * Edit configuration of a plugin
+ * @param {Config} config
  * @param {String} plugin
  * @param {Object} pluginConfig
- * @return {Book}
+ * @return {Config}
  */
-function editPlugin(book, pluginName, pluginConfig) {
-    var config = book.getConfig();
-    config = config.set('pluginsConfig.'+pluginName, pluginConfig);
-
-    return book.setConfig(config);
+function editPlugin(config, pluginName, pluginConfig) {
+    return config.set('pluginsConfig.'+pluginName, pluginConfig);
 }
 
 module.exports = editPlugin;
diff --git a/lib/modifiers/config/removePlugin.js b/lib/modifiers/config/removePlugin.js
index ead22ea..70ec3b1 100644
--- a/lib/modifiers/config/removePlugin.js
+++ b/lib/modifiers/config/removePlugin.js
@@ -1,21 +1,17 @@
 
 /**
  * Remove a plugin from a book's configuration
- * @param {Book} book
+ * @param {Config} config
  * @param {String} plugin
- * @return {Book}
+ * @return {Config}
  */
-function removePlugin(book, pluginName) {
-    var config = book.getConfig();
+function removePlugin(config, pluginName) {
     var deps = config.getPluginDependencies();
 
-
     deps = deps.filter(function(dep) {
         return dep.getName() === pluginName;
     });
-    config = config.setPluginDependencies(deps);
-
-    return book.setConfig(config);
+    return config.setPluginDependencies(deps);
 }
 
 module.exports = removePlugin;
diff --git a/lib/modifiers/config/togglePlugin.js b/lib/modifiers/config/togglePlugin.js
index 13d4406..0cc4932 100644
--- a/lib/modifiers/config/togglePlugin.js
+++ b/lib/modifiers/config/togglePlugin.js
@@ -1,13 +1,12 @@
 
 /**
  * Enable/disable a plugin dependency
- * @param {Book} book
+ * @param {Config} config
  * @param {String} plugin
  * @param {Boolean} state (optional)
- * @return {Book}
+ * @return {Config}
  */
-function togglePlugin(book, plugin, state) {
-    var config = book.getConfig();
+function togglePlugin(config, plugin, state) {
     var deps = config.getPluginDependencies();
 
     deps = deps.map(function(dep) {
@@ -18,8 +17,7 @@ function togglePlugin(book, plugin, state) {
         return dep;
     });
 
-    config = config.setPluginDependencies(deps);
-    return book.setConfig(config);
+    return config.setPluginDependencies(deps);
 }
 
 module.exports = togglePlugin;