improving test case