Add test case && bug fixed