Fix mockery test.