Correcting test while I don't have mocks...