I'll investigate later, but I think we can change this helper with a `mock_model` call