Change config modifier to work on a config object

1 files changed, 4 insertions, 7 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;