diff --git a/tests/test_ilastik/test_ilastik.py b/tests/test_ilastik/test_ilastik.py
index 35a4181368f30119e70055e0f93b4218340f6400..fd6893a55feff793facb931efb6c4265abd9961d 100644
--- a/tests/test_ilastik/test_ilastik.py
+++ b/tests/test_ilastik/test_ilastik.py
@@ -363,7 +363,7 @@ class TestIlastikOnMultichannelInputs(conf.TestServerBaseClass):
         acc_input = generate_file_accessor(self.pa_input_image)
         acc_obmap = generate_file_accessor(res.object_map_filepath)
         self.assertEqual(acc_obmap.hw, acc_input.hw)
-        self.assertEqual(len(acc_obmap._unique()[1]), 3)
+        self.assertEqual(len(acc_obmap.unique()[1]), 3)
 
 
     def test_api(self):
@@ -401,9 +401,9 @@ class TestIlastikObjectClassification(unittest.TestCase):
         stack_ch_pa = stack.get_mono(conf.meta['roiset']['patches_channel'])
         seg_mask = generate_file_accessor(data['multichannel_zstack_mask2d']['path'])
 
-        self.roiset = RoiSet(
+        self.roiset = RoiSet.from_binary_mask(
             stack_ch_pa,
-            get_label_ids(seg_mask),
+            seg_mask,
             params=RoiSetMetaParams(
                 mask_type='boxes',
                 filters={'area': {'min': 1e3, 'max': 1e4}},