improve test case and pass it