model small fix