fixing model more