Update mocks for heka issue #422.