Fix keyvaluestore bugs