fix for testcases