diff --git a/boiling_learning/automl/tuning.py b/boiling_learning/automl/tuning.py
index 29f146d4..16f5992e 100644
--- a/boiling_learning/automl/tuning.py
+++ b/boiling_learning/automl/tuning.py
@@ -39,5 +39,5 @@ def fit_hypermodel(
 
     return TuneModelReturn(
         model=model,
-        evaluation=model.evaluate(ds_val),
+        evaluation=model.evaluate(ds_val.batch(params.batch_size)),
     )