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