diff --git a/glue/core/tests/test_edit_subset_mode.py b/glue/core/tests/test_edit_subset_mode.py
index 1249cf180..3875ecb47 100644
--- a/glue/core/tests/test_edit_subset_mode.py
+++ b/glue/core/tests/test_edit_subset_mode.py
@@ -195,7 +195,8 @@ class Listener(HubListener):
 
     def handler(msg):
         assert len(data1.subsets) == len(data2.subsets)
-        assert data2.subsets[0].subset_state is data1.subsets[0].subset_state
+        for i in range(len(data1.subsets)):
+            assert data2.subsets[i].subset_state is data1.subsets[i].subset_state
         count[0] += 1
 
     app.session.hub.subscribe(listener, SubsetCreateMessage, handler=handler)