diff --git a/roles/proxmox_upgrade/tasks/restore.yml b/roles/proxmox_upgrade/tasks/restore.yml
index bc5c3a7..954ce3e 100644
--- a/roles/proxmox_upgrade/tasks/restore.yml
+++ b/roles/proxmox_upgrade/tasks/restore.yml
@@ -23,8 +23,8 @@
         node: "{{ migration_target }}"
         vmid: "{{ item.vmid }}"
         migrate: true
-        target_node: "{{ current_node }}"
-        online: "{{ true if item.status == 'running' else false }}"
+        target: "{{ current_node }}"
+        
         timeout: "{{ vm_shutdown_timeout }}"
       loop: "{{ kvm_guests | default([]) }}"
       delegate_to: localhost
@@ -41,5 +41,4 @@
 
     - name: "Restore | Complete"
       ansible.builtin.debug:
-        msg: "All guests restored to {{ current_node }}"
-        
\ No newline at end of file
+        msg: "All guests restored to {{ current_node }}"
\ No newline at end of file