Fixing issues with jstree