diff --git a/roles/proxmox_upgrade/tasks/drain.yml b/roles/proxmox_upgrade/tasks/drain.yml
index a193175..60d5baf 100644
--- a/roles/proxmox_upgrade/tasks/drain.yml
+++ b/roles/proxmox_upgrade/tasks/drain.yml
@@ -91,8 +91,7 @@
         node: "{{ current_node }}"
         vmid: "{{ item.vmid }}"
         migrate: true
-        target_node: "{{ migration_target }}"
-        online: "{{ true if item.status == 'running' else false }}"
+        target: "{{ migration_target }}"
         timeout: "{{ vm_shutdown_timeout }}"
       loop: "{{ kvm_guests }}"
       delegate_to: localhost
@@ -108,8 +107,7 @@
         node: "{{ current_node }}"
         vmid: "{{ item.vmid }}"
         migrate: true
-        target_node: "{{ migration_target }}"
-        online: "{{ true if item.status == 'running' else false }}"
+        target: "{{ migration_target }}"
         timeout: "{{ vm_shutdown_timeout }}"
       loop: "{{ kvm_guests }}"
       delegate_to: localhost