Minor bug fix on toggle button wire-up.