Handle mocks with different arguments