Fixed CID model overfitting bug