diff --git a/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php b/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php
index 4a6fd33a6..a94cb6d33 100644
--- a/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php
+++ b/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php
@@ -46,7 +46,7 @@ public function runInference(): MaskProcessingService
             'idCT' => $idCT
         ];
 
-        $inferenceResponse = $this->gaelOProcessingService->executeInference('unet_model', $inferencePayload);
+        $inferenceResponse = $this->gaelOProcessingService->executeInference('pt_seg_attentionunet', $inferencePayload);
         $maskId = $inferenceResponse['id_mask'];
         $maskProcessingService = new MaskProcessingService($this->orthancService, $this->gaelOProcessingService);
         $maskProcessingService->setMaskId($maskId);