diff --git a/src/plugins/collapse-groups/plugin.js b/src/plugins/collapse-groups/plugin.js
index 384bb5f1..6b68c2a7 100644
--- a/src/plugins/collapse-groups/plugin.js
+++ b/src/plugins/collapse-groups/plugin.js
@@ -25,7 +25,7 @@ QueryBuilder.define('collapse-groups', function(options) {
 
     // Collapse any groups that were saved as collapsed
     this.on('afterSetRules', function() {
-        $.each($(Selectors.group_container), function(i, el) {
+        $.each(self.$el.find(Selectors.group_container), function(i, el) {
             var group = self.getModel($(el));
             if (group.collapsed) {
                 self.collapse($(el).find('[data-collapse="group"]:first'), options);