Fixed a unit test fail.