diff --git a/tests/test_atlasq_ohi.py b/tests/test_atlasq_ohi.py
index 96c61572fd841e6a453cde97343d610a52891dce..c4a0938d5f5e6480411a21d3d77861681ee3fdcb 100644
--- a/tests/test_atlasq_ohi.py
+++ b/tests/test_atlasq_ohi.py
@@ -155,8 +155,8 @@ def test_mask(seed):
         exp           = []
 
         for lbl, prop in zip(labels, props):
-            exp.append('{}:{:0.4f}'.format(desc.labels[int(lbl)].name,
-                                           prop))
+            name = desc.find(value=int(lbl)).name
+            exp.append('{}:{:0.4f}'.format(name, prop))
 
         return '\n'.join(exp)