[FIX] KeySet to be thread safe