diff --git a/timdex_dataset_api/__init__.py b/timdex_dataset_api/__init__.py
index e2da0b5..9799cd4 100644
--- a/timdex_dataset_api/__init__.py
+++ b/timdex_dataset_api/__init__.py
@@ -1,13 +1,15 @@
 """timdex_dataset_api/__init__.py"""
 
 from timdex_dataset_api.dataset import TIMDEXDataset
+from timdex_dataset_api.embeddings import TIMDEXEmbeddings
 from timdex_dataset_api.metadata import TIMDEXDatasetMetadata
 from timdex_dataset_api.record import DatasetRecord
 
-__version__ = "3.6.0"
+__version__ = "3.6.1"
 
 __all__ = [
     "DatasetRecord",
     "TIMDEXDataset",
     "TIMDEXDatasetMetadata",
+    "TIMDEXEmbeddings",
 ]
diff --git a/timdex_dataset_api/embeddings.py b/timdex_dataset_api/embeddings.py
index 56e2213..0290896 100644
--- a/timdex_dataset_api/embeddings.py
+++ b/timdex_dataset_api/embeddings.py
@@ -63,8 +63,8 @@ class DatasetEmbedding:
         converter=datetime_iso_parse,
         default=attrs.Factory(lambda: datetime.now(tz=UTC).isoformat()),
     )
-    embedding_vector: list[float] = field(default=None)
-    embedding_object: bytes = field(default=None)
+    embedding_vector: list[float] | None = field(default=None)
+    embedding_object: bytes | None = field(default=None)
 
     @property
     def year(self) -> str: