use get_display_unit

spacetelescope · Oct 4, 2024 · b1ea5af · b1ea5af
1 parent b84ebec
commit b1ea5af
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/jdaviz/configs/cubeviz/plugins/moment_maps/moment_maps.py b/jdaviz/configs/cubeviz/plugins/moment_maps/moment_maps.py
@@ -341,9 +341,7 @@ def moment_zero_unit(self):
 
     @property
     def spectral_unit_selected(self):
-        # NOTE: should we add this logic to app._get_display_unit?
-        unit_conv = self.app.get_tray_item_from_name('g-unit-conversion')
-        return unit_conv.spectral_unit.selected
+        return self.app._get_display_unit('spectral')
 
     def vue_calculate_moment(self, *args):
         self.calculate_moment(add_data=True)