Fixing test case issue