Fix checkbox problem