fix for button bug