diff --git a/jdaviz/tests/test_app.py b/jdaviz/tests/test_app.py
index 3188a01d0b..d210060d9f 100644
--- a/jdaviz/tests/test_app.py
+++ b/jdaviz/tests/test_app.py
@@ -224,13 +224,15 @@ def test_to_unit(cubeviz_helper):
 
     cid = cubeviz_helper.app.data_collection[0].data.find_component_id('flux')
     data = cubeviz_helper.app.data_collection[-1].data
-    values = 1
+    values = [1]
     original_units = u.MJy / u.sr
     target_units = u.MJy
 
     value = uc.to_unit(cubeviz_helper, data, cid, values, original_units, target_units)
 
-    assert np.allclose(value, 4.7945742429049767e-11)
+    # will be a uniform array since not wavelength dependent
+    # so test first value in array
+    assert np.allclose(value[0], 4.800000041882413e-08)
 
     original_units = u.MJy
     target_units = u.MJy / u.sr