diff --git a/scripts/memory/__init__.py b/scripts/memory/__init__.py
index dacb05b3..99523079 100644
--- a/scripts/memory/__init__.py
+++ b/scripts/memory/__init__.py
@@ -37,6 +37,6 @@ def get_memory(cfg, init=False):
 __all__ = [
     "get_memory",
     "LocalCache",
-    "RedisCache",
-    "PineconeCache",
+    "RedisMemory",
+    "PineconeMemory",
 ]