diff --git a/src/otx/algorithms/visual_prompting/tasks/openvino.py b/src/otx/algorithms/visual_prompting/tasks/openvino.py
index ada64d3c937..6f50ef97940 100644
--- a/src/otx/algorithms/visual_prompting/tasks/openvino.py
+++ b/src/otx/algorithms/visual_prompting/tasks/openvino.py
@@ -491,7 +491,7 @@ def forward_decoder(  # type: ignore
                 )
                 has_mask_input = self.has_mask_inputs[0]
 
-            elif is_cascade and i == 1:
+            elif i == 1:
                 # Cascaded Post-refinement-1
                 mask_input, masks = self._postprocess_masks(masks, logits, scores, is_single=True)  # noqa: F821
                 if masks.sum() == 0:
@@ -499,7 +499,7 @@ def forward_decoder(  # type: ignore
 
                 has_mask_input = self.has_mask_inputs[1]
 
-            elif is_cascade and i == 2:
+            elif i == 2:
                 # Cascaded Post-refinement-2
                 mask_input, masks = self._postprocess_masks(masks, logits, scores)  # noqa: F821
                 if masks.sum() == 0: