fix for test case issue