Pooja - #2060 - Fixing test