unit test failure fix