Changing unit tests to use MOQ