diff --git a/kraken/lib/train.py b/kraken/lib/train.py
index fa7dd5450..1ba45eb00 100644
--- a/kraken/lib/train.py
+++ b/kraken/lib/train.py
@@ -839,8 +839,7 @@ def __init__(self,
             train_set.add(page)
 
         if evaluation_data:
-            val_set = BaselineSet(evaluation_data,
-                                  line_width=self.hparams.hyper_params['line_width'],
+            val_set = BaselineSet(line_width=self.hparams.hyper_params['line_width'],
                                   im_transforms=transforms,
                                   augmentation=False,
                                   valid_baselines=valid_baselines,