Checkbox bug fix