This logic doesn't belong here!! Are you fixing a failing test or missing default handling by fixing the implementation in the test code?