Add state assertions