Fix one more error in the unit tests.