add support for on_leave_state feature