DAR-2553 | Fixed unit test