This makes it assert that the function is called with the
intended parameter.

Might fix flakiness of this test.