Estou usando o VS2010 B2 e o EF4 B2 e tentando usar o Rhino Mocks para simular o contexto de entidade gerado pelo EEF.

var context = MockRepository.GenerateMock<SomeDBEntities>();
IObjectSet<TxMode> objectSet = new List<TxMode> { mode }.AsObjectSet();
context.Expect(c => c.TxModes).Return(objectSet);

O problema é quec.TxModes é uma propriedade do tipoObjectSet<TxMode> enquanto eu estou tentando devolver umIObjectSet<TxMode> e recebo o seguinte erro de compilação:

Error 4 The best overloaded method match for 

'Rhino.Mocks.Interfaces.IMethodOptions<System.Data.Objects.ObjectSet
<Data.Entities.TxMode>>.Return(System.Data.Objects.ObjectSet
<Data.Entities.TxMode>)' has some invalid arguments.

O método de extensão AsObjectSet vem deAqui.

Alguma ideia de como posso conseguir isso para compilar?