diff --git a/tests/unit/algorithms/detection/adapters/mmdet/models/detectors/test_custom_dino_detector.py b/tests/unit/algorithms/detection/adapters/mmdet/models/detectors/test_custom_dino_detector.py
index 22b89435f09..2bd02b25505 100644
--- a/tests/unit/algorithms/detection/adapters/mmdet/models/detectors/test_custom_dino_detector.py
+++ b/tests/unit/algorithms/detection/adapters/mmdet/models/detectors/test_custom_dino_detector.py
@@ -34,7 +34,7 @@ def test_custom_dino_load_state_pre_hook(self, fxt_cfg_custom_dino: Dict):
             "memory_trans_fc": "memory_trans_fc",
             "memory_trans_norm": "memory_trans_norm",
         }
-        model.load_state_dict_pre_hook(model, ckpt_dict)
+        model.load_state_dict_pre_hook(ckpt_dict)
 
         assert ckpt_dict["bbox_head.transformer.level_embeds"] == "level_embed"
         assert ckpt_dict["bbox_head.transformer.encoder.attentions.0"] == "encoder.self_attn"