diff --git a/predtuner/pipedbin.py b/predtuner/pipedbin.py
index 18237d1bb1c5a43109648847ea4cb0c5eb276a54..069b42c0913c294240a1b2d6fa04f093611fb90a 100644
--- a/predtuner/pipedbin.py
+++ b/predtuner/pipedbin.py
@@ -101,15 +101,17 @@ class PipedBinaryApp(ModeledApp):
         return qos, time_end - time_begin
 
     def get_models(self) -> List[Union["IPerfModel", "IQoSModel"]]:
-        p2_path = self.model_storage / "p2.json" if self.model_storage else None
+        p1_storage = self.model_storage / "p1.pkl" if self.model_storage else None
+        p2_storage = self.model_storage / "p2.json" if self.model_storage else None
         return [
             LinearPerfModel(self, self.op_costs, self.knob_speedups),
             QoSModelP1(
                 self,
                 lambda conf: self._run_on_knobs(conf, False)[0],
                 self._compute_accuracy,
+                p1_storage
             ),
-            QoSModelP2(self, p2_path),
+            QoSModelP2(self, p2_storage),
         ]
 
     def _compute_accuracy(self, output_tensor: torch.Tensor) -> float: