Fix incorrect button references.
Fix incorrect button references.