Fixed a broken unit test