Check against component_ids rather than doing our own list comprehension

spacetelescope · Jan 2, 2025 · 0c0b470 · 0c0b470
1 parent 9ebdd19
commit 0c0b470
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lcviz/components/components.py b/lcviz/components/components.py
@@ -223,7 +223,7 @@ def _on_change_selected(self, *args):
         # manipulate the arrays in the data-collection directly, and modify FLUX_ORIGIN so that
         # exporting back to a lightkurve object works as expected
         self.app._jdaviz_helper._set_data_component(dc_item, 'flux', dc_item[self.selected])
-        if self.selected+"_err" in [c.label for c in dc_item.components]:
+        if self.selected+"_err" in dc_item.component_ids():
             self.app._jdaviz_helper._set_data_component(dc_item, 'flux_err',
                                                         dc_item[self.selected+"_err"])  # noqa
         else: