Fixing mocks in notification factory test