Add tests for checkbox removal