[IMP]:Improve test case