Added a little bit of diag to modelfix