Fixed bug with check button