fix for the test case