Fixed a bug with lit/unlit button