diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py
index 21a28f3..3f9729a 100644
--- a/neural_lam/models/ar_model.py
+++ b/neural_lam/models/ar_model.py
@@ -461,7 +461,7 @@ def plot_examples(self, batch, n_examples, prediction=None):
                     vis.plot_prediction(
                         pred=pred_t[:, var_i],
                         target=target_t[:, var_i],
-                        datastore=self.datastore,
+                        datastore=self._datastore,
                         title=f"{var_name} ({var_unit}), "
                         f"t={t_i} ({self._datastore.step_length * t_i} h)",
                         vrange=var_vrange,