diff --git a/tools/convert_pkl_to_pb.py b/tools/convert_pkl_to_pb.py
index bc61938..ae2b94e 100644
--- a/tools/convert_pkl_to_pb.py
+++ b/tools/convert_pkl_to_pb.py
@@ -333,7 +333,7 @@ def _save_models(all_net, all_init_net, args):
 
 
 def load_model(args):
-    model = test_engine.initialize_model_from_cfg()
+    model = test_engine.initialize_model_from_cfg(cfg.TEST.WEIGHTS)
     blobs = mutils.get_ws_blobs()
 
     return model, blobs