fix reg model_out