diff --git a/point_e/diffusion/k_diffusion.py b/point_e/diffusion/k_diffusion.py
index 385486d..35f642d 100644
--- a/point_e/diffusion/k_diffusion.py
+++ b/point_e/diffusion/k_diffusion.py
@@ -93,7 +93,7 @@ def sigma_to_t(self, sigma):
         elif alpha_cumprod <= self.diffusion.alphas_cumprod[-1]:
             return self.diffusion.num_timesteps - 1
         else:
-            return float(self.alpha_cumprod_to_t(alpha_cumprod))
+            return int(self.alpha_cumprod_to_t(alpha_cumprod))
 
     def denoise(self, x_t, sigmas, clip_denoised=True, model_kwargs=None):
         t = th.tensor(