swapnil/tushar/sonam: Fixed one broken unit test