Update test to account for randomness