Fix UT failure.

Not sure why, but this test just started failing. Fix it.