diff --git a/pyramid_blending.py b/pyramid_blending.py
index afc4bac..4d13b0b 100644
--- a/pyramid_blending.py
+++ b/pyramid_blending.py
@@ -10,8 +10,8 @@
     build_pyramid,
     find_next_powerof_two,
     is_powerof_two,
-    pad,
 )
+pad = F.pad
 from torch import Tensor
 
 from .nodes_registry import comfy_node