Finally got the mocks working!