Migration of keys.set() to async versions