Fix AI discarding