[IMP] Improved the test case.