Fixed: Broken unit test