One more fix for the test.