little fix on models