fixed a bad bug
adding model parameter