Adjusted the mocks so the specs run again.