fix(button): handle  models