fix button click display logic