fixed failed unit test