one more little change to the model