Fixed a unit test break.