diff --git a/comfy_extras/nodes_latent.py b/comfy_extras/nodes_latent.py
index 952db753..af273681 100644
--- a/comfy_extras/nodes_latent.py
+++ b/comfy_extras/nodes_latent.py
@@ -1,4 +1,5 @@
 import comfy.utils
+import comfy_extras.nodes_post_processing
 import torch
 
 def reshape_latent_to(target_shape, latent):