diff --git a/tests/inference_unittest.py b/tests/inference_unittest.py
index b45348b6ea8dc4ad37889f0706395e31d493c312..c730ef43c9ee33cddc52693e44ddc1e80d7e8c79 100644
--- a/tests/inference_unittest.py
+++ b/tests/inference_unittest.py
@@ -24,7 +24,7 @@ class InferenceTest(DecloudTest):
         system.basic_logging_init()
 
         # Baseline
-        baseline_path = self.get_path("baseline/reconstructed_baseline_w_mosaic.tif")
+        baseline_path = self.get_path("baseline/reconstructed_baseline.tif")
 
         # Model
         model_path = self.get_path("models/crga_os2_david_116-migration-tensorflow-2_466a1_bt32_bv32_lr7e-05_40gpus")