fixing ui model