Enhance tests for checklist state