Dileep/DC | #2952 | Fixing tests