Fix checkbox ui