Use Moq instead of an internal class