Hotfix for testcase

Hotfix for testcase