Fixed getMock to get mocks properly