diff --git a/modules/quanthub_visualization/src/Plugin/Field/FieldFormatter/QuanthubVisualizationFormatter.php b/modules/quanthub_visualization/src/Plugin/Field/FieldFormatter/QuanthubVisualizationFormatter.php
index baf8d5e5..6b1d5918 100644
--- a/modules/quanthub_visualization/src/Plugin/Field/FieldFormatter/QuanthubVisualizationFormatter.php
+++ b/modules/quanthub_visualization/src/Plugin/Field/FieldFormatter/QuanthubVisualizationFormatter.php
@@ -36,7 +36,7 @@ public function viewElements(FieldItemListInterface $items, $langcode): array {
     $dataFilters = $entity->field_visualization_filters->value;
 
     $referencedDataset = $entity
-      ->get('field_qh_visualization_dataset')
+      ->get('field_dataset')
       ->first()
       ->get('entity')
       ->getTarget()
diff --git a/modules/quanthub_visualization/src/Service/PreviewHelper.php b/modules/quanthub_visualization/src/Service/PreviewHelper.php
index fd794141..e375b1fb 100644
--- a/modules/quanthub_visualization/src/Service/PreviewHelper.php
+++ b/modules/quanthub_visualization/src/Service/PreviewHelper.php
@@ -30,7 +30,7 @@ public function __construct(protected EntityTypeManagerInterface $entityTypeMana
    */
   public function provideDrupalSettingsData(FormStateInterface $form_state) {
     $nodeStorage = $this->entityTypeManager->getStorage('node');
-    $datasetUrn = $nodeStorage->load($form_state->getValue('field_qh_visualization_dataset')[0]['target_id'])->get('field_quanthub_urn')->getString();
+    $datasetUrn = $nodeStorage->load($form_state->getValue('field_dataset')[0]['target_id'])->get('field_quanthub_urn')->getString();
 
     $filters = $form_state->getValue('field_qh_visualization_filters')[0]['value'];