Fix button interface issue