Consider using `mock.ANY` for small and icon, as they're not functionally important here.