diff --git a/unit_tests/test_path.py b/unit_tests/test_path.py
index 34843a6..75e3586 100644
--- a/unit_tests/test_path.py
+++ b/unit_tests/test_path.py
@@ -42,7 +42,7 @@ def test_search_path_simple(self):
             dtype=int,
         )
         rel_cost = -rel_cost
-        fcost, fpath = find_most_probably_sequence(
+        fcost, fpath, min_costs_path = find_most_probably_sequence(
             cost,
             region_rel_cost=rel_cost,
             regions=[0, 3],
diff --git a/unit_tests/test_postproc.py b/unit_tests/test_postproc.py
index ac9b7d9..60f4f5f 100644
--- a/unit_tests/test_postproc.py
+++ b/unit_tests/test_postproc.py
@@ -16,7 +16,7 @@ class Test_Post_Processing(unittest.TestCase):
     def test_phase_postprocess(self):
         mri, subreg, vert, label = get_test_mri()
         print(vert.unique())
-        subreg_cleaned, vert_cleaned = phase_postprocess_combined(subreg, vert, debug_data={})
+        subreg_cleaned, vert_cleaned = phase_postprocess_combined(mri, subreg, vert, model_labeling=None, debug_data={})
         self.assertTrue(subreg_cleaned.assert_affine(other=vert_cleaned))
 
         vert_labels = vert_cleaned.unique()