Fix `customButton` function.