diff --git a/ads/feature_store/dataset.py b/ads/feature_store/dataset.py
index 834cbc170..e505a8a76 100644
--- a/ads/feature_store/dataset.py
+++ b/ads/feature_store/dataset.py
@@ -865,6 +865,8 @@ def _update_from_oci_dataset_model(self, oci_dataset: OCIDataset) -> "Dataset":
                         features_list.append(output_feature)
 
                     value = {self.CONST_ITEMS: features_list}
+                elif infra_attr == self.CONST_FEATURE_GROUP:
+                    value = getattr(self.oci_dataset, dsc_attr)
                 else:
                     value = dataset_details[infra_attr]
                 self.set_spec(infra_attr, value)