Fixed unit test failure