fix button that was breaking tests