diff --git a/core/database/classes/__init__.py b/core/database/classes/__init__.py
index ae5adfb..295e6f2 100644
--- a/core/database/classes/__init__.py
+++ b/core/database/classes/__init__.py
@@ -1,4 +1,5 @@
 from __future__ import annotations
 
+from .dataset import Dataset
 from .exceptions import NoDocumentFoundException
 from .visual_communication import VisualCommunication