diff --git a/src/otx/algo/detection/detectors/dfine/hybrid_encoder.py b/src/otx/algo/detection/detectors/dfine/hybrid_encoder.py
index b796281113..b5a99b7c72 100644
--- a/src/otx/algo/detection/detectors/dfine/hybrid_encoder.py
+++ b/src/otx/algo/detection/detectors/dfine/hybrid_encoder.py
@@ -351,7 +351,7 @@ def build_2d_sincos_position_embedding(
 
         return torch.concat([out_w.sin(), out_w.cos(), out_h.sin(), out_h.cos()], dim=1)[None, :, :]
 
-    def forward(self, feats: torch.Tensor) -> list[torch.Tensor]:
+    def forward(self, feats: Tensor) -> list[Tensor]:
         """Forward pass."""
         if len(feats) != len(self.in_channels):
             msg = f"Input feature size {len(feats)} does not match the number of input channels {len(self.in_channels)}"