remodelled the test case