[llvm] [LAA] Fix non-NFC parts of 1aded51 (PR #160701)

[Prabhu Rajasekaran via llvm-commits]

Prabhuk wrote:

We are starting to see the same assertion failure in our builders at Google.

```
```

https://github.com/llvm/llvm-project/pull/160701