diff --git a/latent-upscale-by/latent_upscale_by.py b/latent-upscale-by/latent_upscale_by.py
index 5b9002f..108eeba 100644
--- a/latent-upscale-by/latent_upscale_by.py
+++ b/latent-upscale-by/latent_upscale_by.py
@@ -1,5 +1,5 @@
 import comfy.utils
-
+import math
 
 class LatentUpscaleBy:
     upscale_methods = ["nearest-exact", "bilinear", "area"]
@@ -18,7 +18,11 @@ class LatentUpscaleBy:
     def upscale(self, samples, upscale_method, scale, crop):
         s = samples.copy()
         w = round(samples["samples"].shape[3] * 8 * scale)
+        w = 64 * math.ceil(w / 64)
+
         h = round(samples["samples"].shape[2] * 8 * scale)
+        h = 64 * math.ceil(h / 64)
+        
         s["samples"] = comfy.utils.common_upscale(
             samples["samples"], w // 8, h // 8, upscale_method, crop)
         return (s,)