Chaning unit test to use Mock.Setup instead of setting it directly