adding test case correction