correcting how the model works