Fixed a broken unit test.