Please fix the tests for this case, possibly modifying the mock will be enough