Fix undefined button