diff --git a/comfy_api/latest/_io.py b/comfy_api/latest/_io.py
index c527190ec..ce7dbb434 100644
--- a/comfy_api/latest/_io.py
+++ b/comfy_api/latest/_io.py
@@ -96,8 +96,8 @@ def comfytype(io_type: str, **kwargs):
 
     A ComfyType may have the following attributes:
     - Type = <type hint here>
-    - class Input(InputV3): ...
-    - class Output(OutputV3): ...
+    - class Input(Input): ...
+    - class Output(Output): ...
     '''
     def decorator(cls: T) -> T:
         if isinstance(cls, _ComfyType) or issubclass(cls, _ComfyType):
@@ -737,7 +737,7 @@ class MultiType:
         @property
         def io_types(self) -> list[type[Input]]:
             '''
-            Returns list of InputV3 class types permitted.
+            Returns list of Input class types permitted.
             '''
             io_types = []
             for x in self._io_types:
diff --git a/comfy_api/v0_0_2/__init__.py b/comfy_api/v0_0_2/__init__.py
index 90ca77e42..34de96636 100644
--- a/comfy_api/v0_0_2/__init__.py
+++ b/comfy_api/v0_0_2/__init__.py
@@ -1,6 +1,7 @@
 from comfy_api.latest import ComfyAPI_latest
 from typing import Type, TYPE_CHECKING
 from comfy_api.internal.async_to_sync import create_sync_class
+from comfy_api.latest import io, ui, resources  #noqa: F401
 
 class ComfyAPIAdapter_v0_0_2(ComfyAPI_latest):
     VERSION = "0.0.2"