small fix on models