little fix on button toggle