diff --git a/paperforge/memory/__init__.py b/paperforge/memory/__init__.py
index 97d74fc9..5585cd62 100644
--- a/paperforge/memory/__init__.py
+++ b/paperforge/memory/__init__.py
@@ -1,8 +1,11 @@
 from __future__ import annotations
 
 from paperforge.memory.db import get_connection, get_memory_db_path
+from paperforge.memory.schema import ensure_schema, drop_all_tables
 
 __all__ = [
     "get_connection",
     "get_memory_db_path",
+    "ensure_schema",
+    "drop_all_tables",
 ]