Make on buttons work right.