Fixed update logic to not reify FeatureStore.