Made the first test work with an spy, mocks should come after that.