Fix overfitting