Clean up calls to kv.set/get to use the new kv.store.get/set api

This'll fix the test failures.