diff --git a/jdaviz/configs/default/plugins/subset_tools/subset_tools.py b/jdaviz/configs/default/plugins/subset_tools/subset_tools.py
index 888d68d0a3..74f64dbf71 100644
--- a/jdaviz/configs/default/plugins/subset_tools/subset_tools.py
+++ b/jdaviz/configs/default/plugins/subset_tools/subset_tools.py
@@ -241,7 +241,7 @@ def get_regions(self, region_type=None, list_of_subset_labels=None,
                 subset_data = lyr.layer
                 subset_label = subset_data.label
 
-                if type(subset_data.subset_state) not in (RoiSubsetState, CompositeSubsetState):
+                if isinstance(subset_data.subset_state, MaskSubsetState):
                     # Ignore MaskSubsetState here
                     continue