ForceMock instead?