diff --git a/tests/dragon/test_error_handling.py b/tests/dragon/test_error_handling.py
index e85efb723e..c929c23416 100644
--- a/tests/dragon/test_error_handling.py
+++ b/tests/dragon/test_error_handling.py
@@ -117,7 +117,7 @@ def setup_worker_manager_model_key(test_dir, monkeypatch: pytest.MonkeyPatch):
     )
 
     tensor_key = MessageHandler.build_tensor_key("key", feature_store.descriptor)
-    model_key = MessageHandler.build_model_key("model key")
+    model_key = MessageHandler.build_model_key("model key", feature_store.descriptor)
     request = MessageHandler.build_request(
         test_dir, model_key, [tensor_key], [tensor_key], [], None
     )