fix the UI Button error