Fixed some broken buttons