little fix on models
little fix on models