fix last test case