diff --git a/python/sglang/multimodal_gen/test/unit/test_transformer_loader_fallback.py b/python/sglang/multimodal_gen/test/unit/test_transformer_loader_fallback.py
index f237ec06a..b412a3ed9 100644
--- a/python/sglang/multimodal_gen/test/unit/test_transformer_loader_fallback.py
+++ b/python/sglang/multimodal_gen/test/unit/test_transformer_loader_fallback.py
@@ -40,6 +40,7 @@ class TestTransformerLoaderFallbackAdmission(unittest.TestCase):
             "dp_size": 1,
             "use_fsdp_inference": False,
             "resolve_component_attention_backend": mock.Mock(return_value=(None, None)),
+            "should_direct_gpu_weight_load_component": mock.Mock(return_value=False),
             "should_use_fsdp_for_component": mock.Mock(return_value=fsdp_requested),
         }
         values.update(overrides)