Make unhandled state work better