Quick fix for button bug.