diff --git a/yepcode_run/__init__.py b/yepcode_run/__init__.py
index 1856679..fed7512 100644
--- a/yepcode_run/__init__.py
+++ b/yepcode_run/__init__.py
@@ -2,6 +2,7 @@
 from .run.execution import Execution
 from .api.yepcode_api import YepCodeApi
 from .env.yepcode_env import YepCodeEnv
+from .storage.yepcode_storage import YepCodeStorage
 from .api.types import (
     YepCodeApiConfig,
     ExecutionStatus,
@@ -18,6 +19,7 @@
 __all__ = [
     "YepCodeRun",
     "YepCodeEnv",
+    "YepCodeStorage",
     "Execution",
     "YepCodeApi",
     "YepCodeApiConfig",