fixing bug when loading model run