fix model issue