Before revision:  However, the metric improvements on the validation set may not translate to the test set, especially  on small datasets.
After revision:  However, the metric improvements on the validation set may not translate to the test set, especially when tuning models trained on small datasets.