Ajay/DC | #0000 | Fixing ut after merge