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"
diff --git a/comfy_extras/nodes_v3_test.py b/comfy_extras/nodes_v3_test.py
index 9e43ef290..d4e1762a4 100644
--- a/comfy_extras/nodes_v3_test.py
+++ b/comfy_extras/nodes_v3_test.py
@@ -1,6 +1,7 @@
 import torch
 import time
-from comfy_api.latest import io, ui, resources, _io
+from comfy_api.v0_0_2 import io, ui, resources
+from comfy_api.latest import _io
 import logging  # noqa
 import folder_paths
 import comfy.utils