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