diff --git a/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py b/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py
index 19f17366459a7684c6df8a940438b661cf7f6029..2fb1de17ee226571e6cd6b808640bf35280932db 100644
--- a/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py
+++ b/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py
@@ -39,7 +39,7 @@ for model_cls, nch, img_size, batch_size, pathname in benchmarks:
         params / "test_input.bin", params / "test_labels.bin", dataset_shape
     )
     model: Module = model_cls()
-    checkpoint = self_folder / "../model_params" / f"{pathname}.pth.tar"
+    checkpoint = self_folder / "../model_params/pytorch" / f"{pathname}.pth.tar"
     model.load_state_dict(torch.load(checkpoint.as_posix()))
 
     build_dir = codegen_dir / "build"