corrected bug for button