diff --git a/llama_stack/providers/inline/datasetio/localfs/config.py b/llama_stack/providers/inline/datasetio/localfs/config.py
index 4147ebd8c..ecdcfd743 100644
--- a/llama_stack/providers/inline/datasetio/localfs/config.py
+++ b/llama_stack/providers/inline/datasetio/localfs/config.py
@@ -15,4 +15,4 @@ from llama_stack.providers.utils.kvstore.config import (
 class LocalFSDatasetIOConfig(BaseModel):
     kvstore: KVStoreConfig = SqliteKVStoreConfig(
         db_path=(RUNTIME_BASE_DIR / "localfs_datasetio.db").as_posix()
-    )  # Uses SQLite config specific to HF storage
+    )  # Uses SQLite config specific to localfs storage
diff --git a/llama_stack/providers/inline/datasetio/localfs/datasetio.py b/llama_stack/providers/inline/datasetio/localfs/datasetio.py
index b624a152a..f02401088 100644
--- a/llama_stack/providers/inline/datasetio/localfs/datasetio.py
+++ b/llama_stack/providers/inline/datasetio/localfs/datasetio.py
@@ -89,6 +89,7 @@ class LocalFSDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
         self.config = config
         # local registry for keeping track of datasets within the provider
         self.dataset_infos = {}
+        self.kvstore = None
 
     async def initialize(self) -> None:
         self.kvstore = await kvstore_impl(self.config.kvstore)
@@ -113,11 +114,11 @@ class LocalFSDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
     ) -> None:
         # Store in kvstore
         key = f"{DATASETS_PREFIX}{dataset.identifier}"
-        dataset_impl = PandasDataframeDataset(dataset)
         await self.kvstore.set(
             key=key,
             value=dataset.json(),
         )
+        dataset_impl = PandasDataframeDataset(dataset)
         self.dataset_infos[key] = DatasetInfo(
             dataset_def=dataset,
             dataset_impl=dataset_impl,