diff --git a/adala/memories/base.py b/adala/memories/base.py
index db6376c5..85674d01 100644
--- a/adala/memories/base.py
+++ b/adala/memories/base.py
@@ -3,8 +3,6 @@
 from typing import Any, Optional, TYPE_CHECKING, Dict
 
 from pydantic import BaseModel, Field
-from adala.datasets.base import Dataset, InternalDataFrame
-from rich import print
 
 
 class Memory(BaseModel, ABC):