diff --git a/jsTree.directive.js b/jsTree.directive.js
index c636046..265346d 100644
--- a/jsTree.directive.js
+++ b/jsTree.directive.js
@@ -45,7 +45,11 @@ ngJSTree.directive('jsTree', ['$http', function($http) {
         }
 
         if (config.plugins.indexOf('checkbox') >= 0) {
-          config.checkbox = config.checkbox || {};
+          if (a.treeCheckbox) {
+            config.checkbox = s[a.treeCheckbox];
+          } else {
+            config.checkbox = config.checkbox || {};
+          }
           config.checkbox.keep_selected_style = false;
         }