fixes issue with new button setup.