small fix on model