We need this because `expect_any_instance_of` doesn't work on
prepended models. Now we could use the same code between CE/EE