update UnitTests for model