Fix remaining  flakiness in the test.

In under 1.0% cases test app actually gets sample uniformly, which
is not expected flow of this test.

Test: atest RuntimeMessageCollectionTest
Bug: 151432852
Change-Id: If8d90e9272bc41403d9d2555b02389bee8974909