small fix in model