diff --git a/jdaviz/configs/cubeviz/plugins/moment_maps/tests/test_moment_maps.py b/jdaviz/configs/cubeviz/plugins/moment_maps/tests/test_moment_maps.py index 30692f7a2d..6ad2f5adb2 100644 --- a/jdaviz/configs/cubeviz/plugins/moment_maps/tests/test_moment_maps.py +++ b/jdaviz/configs/cubeviz/plugins/moment_maps/tests/test_moment_maps.py @@ -22,7 +22,7 @@ def test_moment_calculation(cubeviz_helper, spectrum1d_cube, tmpdir): assert mm.moment_available assert dc[1].label == 'moment 0' - mv_data = app.get_viewer('mask-viewer').data() + mv_data = cubeviz_helper.app.get_viewer('mask-viewer').data() # by default, will overwrite the previous entry (so only one data entry) assert len(mv_data) == 1 assert mv_data[0].label == 'moment 0'