Fix the failed test and add one more