Ajay | #2600 - Fixing failing UT test