diff --git a/comfy_extras/nodes_save_npz.py b/comfy_extras/nodes_save_npz.py
index 756a01907..fb9e4c877 100644
--- a/comfy_extras/nodes_save_npz.py
+++ b/comfy_extras/nodes_save_npz.py
@@ -2,7 +2,7 @@ import os
 
 import folder_paths
 from comfy_api.latest import io
-from comfy_api.latest._util.npz_types import NPZ
+from comfy_api_sealed_worker.npz_types import NPZ
 
 
 class SaveNPZ(io.ComfyNode):
diff --git a/comfy_extras/nodes_save_ply.py b/comfy_extras/nodes_save_ply.py
index 6ee7a67e9..64c32e4de 100644
--- a/comfy_extras/nodes_save_ply.py
+++ b/comfy_extras/nodes_save_ply.py
@@ -2,7 +2,7 @@ import os
 
 import folder_paths
 from comfy_api.latest import io
-from comfy_api.latest._util.ply_types import PLY
+from comfy_api_sealed_worker.ply_types import PLY
 
 
 class SavePLY(io.ComfyNode):