DC,Dileep | Fixing unit test