diff --git a/adminsortable/static/adminsortable/js/admin.sortable.stacked.inlines.js b/adminsortable/static/adminsortable/js/admin.sortable.stacked.inlines.js
index 852fe29..3b32b7e 100644
--- a/adminsortable/static/adminsortable/js/admin.sortable.stacked.inlines.js
+++ b/adminsortable/static/adminsortable/js/admin.sortable.stacked.inlines.js
@@ -40,7 +40,9 @@
                         type: 'POST',
                         data: { indexes : indexes.join(',') },
                         success: function() {
-                            ui.item.find('.form-row').effect('highlight', {}, 1000).removeAttr('style');
+                            var fieldsets = ui.item.find('fieldset'),
+                                highlightedSelector = fieldsets.filter('.collapsed').length === fieldsets.length ? 'h3' : '.form-row';
+                            ui.item.find(highlightedSelector).effect('highlight', {}, 1000);
                         }
                     });
                 }
diff --git a/sample_project/app/admin.py b/sample_project/app/admin.py
index 8193e27..a029d62 100644
--- a/sample_project/app/admin.py
+++ b/sample_project/app/admin.py
@@ -14,6 +14,16 @@ admin.site.register(Category, SortableAdmin)
 
 
 class ComponentInline(SortableStackedInline):
+    # fieldsets = (
+    #     ('foo', {
+    #         'classes': ('collapse',),
+    #         'fields': ('title',)
+    #     }),
+    #     ('', {
+    #         'classes': ('collapse',),
+    #         'fields': ('widget',)
+    #     }),
+    # )
     model = Component
 
     def queryset(self, request):