fixed the failed unit test