Fixed unit test that was failing