diff --git a/dynthres_comfyui.py b/dynthres_comfyui.py
index 4f17f39..c8644f4 100644
--- a/dynthres_comfyui.py
+++ b/dynthres_comfyui.py
@@ -70,7 +70,7 @@ class DynamicThresholdingSimpleComfyNode:
             uncond = input - args["uncond"]
             cond_scale = args["cond_scale"]
             time_step = model.model.model_sampling.timestep(args["sigma"])
-            time_step = time_step[0].value()
+            time_step = time_step[0].item()
             dynamic_thresh.step = 999 - time_step
 
             return input - dynamic_thresh.dynthresh(cond, uncond, cond_scale, None)