diff --git a/Dockerfile b/Dockerfile
index 31f4979df227f27eafad185c6527a646c5fdc2be..8bc8ceaa4d6d3148e6218bd54406dec703a42705 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -19,8 +19,8 @@ COPY docker/requirements.txt /tmp/requirements.txt
 RUN python3 -m pip install -r /tmp/requirements.txt
 
 # Build remote modules
-RUN cd /src/otb/otb/Modules/Remote/ && git clone https://gitlab.irstea.fr/remi.cresson/SimpleExtractionTools.git
-RUN cd /src/otb/otb/Modules/Remote/ && git clone https://gitlab.irstea.fr/remi.cresson/mlutils.git
+RUN cd /src/otb/otb/Modules/Remote/ && git clone https://forgemia.inra.fr/orfeo-toolbox/otb-simpleextractiontools.git
+RUN cd /src/otb/otb/Modules/Remote/ && git clone https://forgemia.inra.fr/orfeo-toolbox/otb-mlutils.git
 COPY . /src/otb/otb/Modules/Remote/decloud/
 RUN cd /src/otb/build/OTB/build && cmake /src/otb/otb/ -DModule_SimpleExtractionTools=ON -DModule_MLUtils=ON -DBUILD_TESTING=OFF -DModule_OTBDecloud=ON 
 RUN cd /src/otb/build/OTB/build && make -j $(nproc --all) install
diff --git a/tests/inference_unittest.py b/tests/inference_unittest.py
index 316b9c486c90cc64cd75bea0c11f3f874215d59b..b45348b6ea8dc4ad37889f0706395e31d493c312 100644
--- a/tests/inference_unittest.py
+++ b/tests/inference_unittest.py
@@ -27,7 +27,7 @@ class InferenceTest(DecloudTest):
         baseline_path = self.get_path("baseline/reconstructed_baseline_w_mosaic.tif")
 
         # Model
-        model_path = self.get_path("models/crga_os2david_occitanie_pretrained")
+        model_path = self.get_path("models/crga_os2_david_116-migration-tensorflow-2_466a1_bt32_bv32_lr7e-05_40gpus")
 
         # Input sources
         s1_tm1 = [
@@ -93,7 +93,7 @@ class InferenceTest(DecloudTest):
         baseline_path = self.get_path("baseline/reconstructed_baseline_w_preprocessor.tif")
 
         # Model
-        model_path = self.get_path("models/crga_os2david_occitanie_pretrained")
+        model_path = self.get_path("models/crga_os2_david_116-migration-tensorflow-2_466a1_bt32_bv32_lr7e-05_40gpus")
 
         # Input sources
         s1_tm1 = [