this fixes the button