[BACKLOG-824] - Fixed unit test