diff --git a/tiled_vae.py b/tiled_vae.py
index 20b9bb6..ac2c826 100644
--- a/tiled_vae.py
+++ b/tiled_vae.py
@@ -749,7 +749,7 @@ class VAEHook:
             del result, result_approx
             comfy.model_management.throw_exception_if_processing_interrupted()
         vae_dtype = comfy.model_management.vae_dtype()
-        return result.to(dtype=vae_dtype) if result is not None else result_approx.to(device=device, dtype=vae_dtype)
+        return result.to(dtype=vae_dtype, device=device) if result is not None else result_approx.to(device=device, dtype=vae_dtype)
 
 # from .tiled_vae import VAEHook, get_rcmd_enc_tsize, get_rcmd_dec_tsize
 from nodes import VAEEncode, VAEDecode
@@ -865,4 +865,4 @@ NODE_CLASS_MAPPINGS = {
 NODE_DISPLAY_NAME_MAPPINGS = {
     "VAEEncodeTiled_TiledDiffusion": "Tiled VAE Encode",
     "VAEDecodeTiled_TiledDiffusion": "Tiled VAE Decode",
-}
\ No newline at end of file
+}