Fix the failed test