diff --git a/diffusers_in_comfy.py b/diffusers_in_comfy.py
index 5b256a2..49fb3cd 100644
--- a/diffusers_in_comfy.py
+++ b/diffusers_in_comfy.py
@@ -95,7 +95,7 @@ class Text2ImgStableDiffusionPipeline:
             pipeline.enable_model_cpu_offload()
 
 
-        pipeline.to(device)
+        pipeline = pipeline.to(device)
 
         print(f'Low VRAM options is {low_vram}, sending pipeline to {device}')
         return (pipeline,)
@@ -373,6 +373,7 @@ class LoRALoader:
         pipeline.set_adapters([lora_name], adapter_weights=[lora_scale])
 
 
+
         return (pipeline,)