Adapt to our dataset.