fix of the previous bug in model_2