diff --git a/docling/datamodel/base_models.py b/docling/datamodel/base_models.py
index 78cc50d0..9bf4ee1d 100644
--- a/docling/datamodel/base_models.py
+++ b/docling/datamodel/base_models.py
@@ -5,7 +5,7 @@ from io import BytesIO
 from typing import Annotated, Any, Dict, List, Optional, Tuple, Union
 
 from docling_core.types.experimental import BoundingBox, Size
-from docling_core.types.experimental.document import BaseFigureData, TableCell
+from docling_core.types.experimental.document import BasePictureData, TableCell
 from docling_core.types.experimental.labels import DocItemLabel
 from PIL.Image import Image
 from pydantic import BaseModel, ConfigDict, Field, model_validator
@@ -84,7 +84,7 @@ class TextElement(BasePageElement): ...
 
 
 class FigureElement(BasePageElement):
-    data: Optional[BaseFigureData] = None
+    data: Optional[BasePictureData] = None
     provenance: Optional[str] = None
     predicted_class: Optional[str] = None
     confidence: Optional[float] = None
diff --git a/docling/models/ds_glm_model.py b/docling/models/ds_glm_model.py
index 31690f4c..74fc9879 100644
--- a/docling/models/ds_glm_model.py
+++ b/docling/models/ds_glm_model.py
@@ -57,7 +57,7 @@ class GlmModel:
                     if arr == "tables":
                         prov = ds_document.tables[index].prov[0]
                     elif arr == "figures":
-                        prov = ds_document.figures[index].prov[0]
+                        prov = ds_document.pictures[index].prov[0]
                     else:
                         prov = None