Fix failing test.
Fix failing test.