diff --git a/pts/model/tempflow/tempflow_estimator.py b/pts/model/tempflow/tempflow_estimator.py
index c0ac17f..ff577bb 100644
--- a/pts/model/tempflow/tempflow_estimator.py
+++ b/pts/model/tempflow/tempflow_estimator.py
@@ -188,7 +188,6 @@ class TempFlowEstimator(PTSEstimator):
         prediction_network = TempFlowPredictionNetwork(
             input_size=self.input_size,
             target_dim=self.target_dim,
-            num_parallel_samples=self.num_parallel_samples,
             num_layers=self.num_layers,
             num_cells=self.num_cells,
             cell_type=self.cell_type,