Fix broken buttons