Fix for testcase failure in integration repo