We're passing in the state-object, asking to create the PSO based on
this. Yes, these are the same, but I wouldn't bet on that being the case
forever.

So let's start checking the state-object instead. This should be more
resilient to changes.